andreped commited on
Commit
b19d82f
1 Parent(s): 61aa9bc

Fixed airways task-related naming

Browse files
Files changed (3) hide show
  1. demo/app.py +1 -1
  2. demo/src/compute.py +2 -2
  3. demo/src/gui.py +1 -1
demo/app.py CHANGED
@@ -7,7 +7,7 @@ def main():
7
  # cwd = "/Users/andreped/workspace/AeroPath/" # local testing -> macOS
8
  cwd = "/home/user/app/" # production -> docker
9
 
10
- class_name = "tumor"
11
 
12
  # initialize and run app
13
  app = WebUI(class_name=class_name, cwd=cwd)
 
7
  # cwd = "/Users/andreped/workspace/AeroPath/" # local testing -> macOS
8
  cwd = "/home/user/app/" # production -> docker
9
 
10
+ class_name = "airways"
11
 
12
  # initialize and run app
13
  app = WebUI(class_name=class_name, cwd=cwd)
demo/src/compute.py CHANGED
@@ -8,8 +8,8 @@ def run_model(
8
  input_path: str,
9
  model_path: str,
10
  verbose: str = "info",
11
- task: str = "MRI_Meningioma",
12
- name: str = "Tumor",
13
  ):
14
  logging.basicConfig()
15
  logging.getLogger().setLevel(logging.WARNING)
 
8
  input_path: str,
9
  model_path: str,
10
  verbose: str = "info",
11
+ task: str = "CT_Airways",
12
+ name: str = "Airways",
13
  ):
14
  logging.basicConfig()
15
  logging.getLogger().setLevel(logging.WARNING)
demo/src/gui.py CHANGED
@@ -74,7 +74,7 @@ class WebUI:
74
 
75
  with gr.Row():
76
  gr.Examples(
77
- examples=[self.cwd + "lung_001.nii.gz"],
78
  inputs=file_output,
79
  outputs=file_output,
80
  fn=self.upload_file,
 
74
 
75
  with gr.Row():
76
  gr.Examples(
77
+ examples=[self.cwd + "test_thorax_CT.nii.gz"],
78
  inputs=file_output,
79
  outputs=file_output,
80
  fn=self.upload_file,