Merge pull request #4 from andreped/demo

Fixed airways task-related naming
raidionics · Oct 3, 2023 · 5dd2b05 · 5dd2b05
2 parents d2434ad + b19d82f
commit 5dd2b05
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/demo/app.py b/demo/app.py
@@ -7,7 +7,7 @@ def main():
     # cwd = "/Users/andreped/workspace/AeroPath/"  # local testing -> macOS
     cwd = "/home/user/app/"  # production -> docker
 
-    class_name = "tumor"
+    class_name = "airways"
 
     # initialize and run app
     app = WebUI(class_name=class_name, cwd=cwd)

diff --git a/demo/src/compute.py b/demo/src/compute.py
@@ -8,8 +8,8 @@ def run_model(
     input_path: str,
     model_path: str,
     verbose: str = "info",
-    task: str = "MRI_Meningioma",
-    name: str = "Tumor",
+    task: str = "CT_Airways",
+    name: str = "Airways",
 ):
     logging.basicConfig()
     logging.getLogger().setLevel(logging.WARNING)

diff --git a/demo/src/gui.py b/demo/src/gui.py
@@ -74,7 +74,7 @@ def run(self):
 
             with gr.Row():
                 gr.Examples(
-                    examples=[self.cwd + "lung_001.nii.gz"],
+                    examples=[self.cwd + "test_thorax_CT.nii.gz"],
                     inputs=file_output,
                     outputs=file_output,
                     fn=self.upload_file,