From b19d82f65c608350cabe6754dd0b4b8906a04e44 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andr=C3=A9=20Pedersen?= <andrped94@gmail.com>
Date: Tue, 3 Oct 2023 16:52:38 +0200
Subject: [PATCH] Fixed airways task-related naming

---
 demo/app.py         | 2 +-
 demo/src/compute.py | 4 ++--
 demo/src/gui.py     | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/demo/app.py b/demo/app.py
index d68b476..f8f2245 100644
--- 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
index a5df82b..305396f 100644
--- 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
index 807babb..a606479 100644
--- 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,