andreped commited on
Commit
1e90a57
1 Parent(s): fec5809

Minor refactoring

Browse files
Files changed (2) hide show
  1. demo/src/gui.py +0 -1
  2. 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