Update app.py

app.py

@@ -81,7 +81,7 @@ def infer_mediar(image, temp_dir="temp_mediar"):
     }
 
     model = MEDIARFormer(**model_args)
-    weights = torch.load("MEDIAR_Weights/from_phase1.pth", map_location=device)
+    weights = torch.load("from_phase1.pth", map_location=device)
     model.load_state_dict(weights, strict=False)
     model.to(device)
     model.eval()
@@ -177,7 +177,7 @@ with gr.Blocks(title="Cell Segmentation Explorer") as app:
     gr.Markdown("---")
     gr.Markdown("### Sample Images (click to use as input)")
 
-    original_sample_paths = ["Sample Images/img1.png", "Sample Images/img2.png", "Sample Images/img3.png"]
+    original_sample_paths = ["img1.png", "img2.png", "img3.png"]
     resized_sample_paths = []
 
     for idx, p in enumerate(original_sample_paths):