Spaces:
Running
Running
Minor refactoring
Browse files- demo/src/gui.py +0 -1
- demo/src/inference.py +1 -0
demo/src/gui.py
CHANGED
|
@@ -11,7 +11,6 @@ from .logger import setup_logger
|
|
| 11 |
from .utils import load_ct_to_numpy
|
| 12 |
from .utils import load_pred_volume_to_numpy
|
| 13 |
|
| 14 |
-
|
| 15 |
# setup logging
|
| 16 |
LOGGER = setup_logger()
|
| 17 |
|
|
|
|
| 11 |
from .utils import load_ct_to_numpy
|
| 12 |
from .utils import load_pred_volume_to_numpy
|
| 13 |
|
|
|
|
| 14 |
# setup logging
|
| 15 |
LOGGER = setup_logger()
|
| 16 |
|
demo/src/inference.py
CHANGED
|
@@ -91,6 +91,7 @@ def run_model(
|
|
| 91 |
shutil.rmtree(patient_directory)
|
| 92 |
if os.path.exists(output_path):
|
| 93 |
shutil.rmtree(output_path)
|
|
|
|
| 94 |
except Exception:
|
| 95 |
print(traceback.format_exc())
|
| 96 |
# Clean-up
|
|
|
|
| 91 |
shutil.rmtree(patient_directory)
|
| 92 |
if os.path.exists(output_path):
|
| 93 |
shutil.rmtree(output_path)
|
| 94 |
+
|
| 95 |
except Exception:
|
| 96 |
print(traceback.format_exc())
|
| 97 |
# Clean-up
|