Update app.py

app.py

@@ -216,7 +216,7 @@ def run_diagnosis(backbone_name, input_image, preprocess_fn = None, Idx2labels=N
     input_tensor = input_tensor.unsqueeze(dim = 0)
     # newimg = transform(img).unsqueeze(dim=0)
 
-    CKPT_PATH = os.path.join(os.getcwd(), f"{backbone_name}.ckpt")
+    CKPT_PATH = os.path.join(CKPT_ROOT, f"{backbone_name}.ckpt")
     model = modelModule.load_from_checkpoint(CKPT_PATH)
     model.eval()
 
@@ -237,6 +237,7 @@ def run_diagnosis(backbone_name, input_image, preprocess_fn = None, Idx2labels=N
     return predicted_classes, output_probs
 
 """### Gradio app"""
+CKPT_ROOT = os.path.join(os.getcwd(), "Trained models")
 
 example_list_dir = os.path.join(os.getcwd(), "Curated test samples")
 example_list_img_names = os.listdir(example_list_dir)