k

app.py

@@ -19,7 +19,12 @@ def load_models():
     global sam_model, sd_pipe, predictor
     if sam_model is None:
         print("Loading SAM model...")
-        sam_model = sam_model_registry["vit_h"](checkpoint="sam_vit_b_01ec64.pth")
+        sam_model = sam_model_registry["vit_b"](checkpoint="sam_vit_b_01ec64.pth")  # Use available checkpoint
+        missing_keys, unexpected_keys = sam_model.load_state_dict(
+            torch.load("sam_vit_b_01ec64.pth"), strict=False
+        )
+        if missing_keys or unexpected_keys:
+            print(f"Warning: Missing keys: {missing_keys}, Unexpected keys: {unexpected_keys}")
         predictor = SamPredictor(sam_model)
     
     if sd_pipe is None: