Update app.py

app.py

@@ -10,7 +10,13 @@ import os
 
 # Load SAM and MiDaS models
 def load_models():
+
+ 
     sam_checkpoint = "sam_vit_b_01ec64.pth"
+    model_type = "vit_b"  # <-- Must match checkpoint
+
+    sam = sam_model_registry[model_type](checkpoint=sam_checkpoint).to(device)
+
     if not os.path.exists(sam_checkpoint):
         raise FileNotFoundError("Please upload the SAM checkpoint file to the working directory.")