From 217907a16b77ec6f199720d1c35e7f984ea34c08 Mon Sep 17 00:00:00 2001 From: Ravnoor Gill Date: Thu, 14 Sep 2023 11:50:16 -0400 Subject: [PATCH] add conditional import for tensorflow --- app/utils/image_processing.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/app/utils/image_processing.py b/app/utils/image_processing.py index 29fc947..cf0e2fb 100644 --- a/app/utils/image_processing.py +++ b/app/utils/image_processing.py @@ -8,8 +8,8 @@ import multiprocessing import numpy as np # import zipfile - -from antspynet.utilities import brain_extraction +if os.environ.get("BRAIN_MASKING") == "cpu": + from antspynet.utilities import brain_extraction from .deepmask import * from .helpers import * from matplotlib.backends.backend_pdf import PdfPages