Upload app.py

app.py

@@ -453,16 +453,23 @@ def run_segformer(img_rgb, morph_cleanup=False):
         paths_to_try = [
             os.path.join(base_dir, "SegFormer_Model", "best_segformer_dice_model"), # Local PC
             "best_segformer_dice_model",                                            # Hugging Face Root
-            os.path.join(os.path.dirname(__file__), "best_segformer_dice_model")    # Next to app.py
+            os.path.join(os.path.dirname(__file__), "best_segformer_dice_model"),   # Next to app.py
         ]
         
+        # If files were uploaded directly to the root (no folder)
+        if os.path.exists("config.json"):
+            paths_to_try.append(".")
+        if os.path.exists(os.path.join(os.path.dirname(__file__), "config.json")):
+            paths_to_try.append(os.path.dirname(__file__))
+            
         model_path = None
         for p in paths_to_try:
-            if os.path.exists(p):
+            # For SegFormer, the path must contain config.json
+            if os.path.exists(p) and os.path.exists(os.path.join(p, "config.json")):
                 model_path = p
                 break
                 
-        # If not found, fallback to Hugging Face standard path just in case
+        # Fallback
         if model_path is None:
             model_path = "best_segformer_dice_model"