Spaces:

lakshmi082024
/

sam_object

Runtime error

lakshmi082024 commited on May 10

Commit

f48d218

verified ·

1 Parent(s): 676f7e2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ import os
 # Load SAM and MiDaS models
 def load_models():
-    sam_checkpoint = "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth"
     if not os.path.exists(sam_checkpoint):
         raise FileNotFoundError("Please upload the SAM checkpoint file to the working directory.")

 # Load SAM and MiDaS models
 def load_models():
+    sam_checkpoint = "sam_vit_b_01ec64.pth"
     if not os.path.exists(sam_checkpoint):
         raise FileNotFoundError("Please upload the SAM checkpoint file to the working directory.")