Spaces:

zht1
/

test2

Runtime error

jianyouli commited on Nov 1, 2023

Commit

fdef182

1 Parent(s): 42403d4

Add application file3

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ from utils.tools_gradio import fast_process
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 # Load the pre-trained model
-sam_checkpoint = r"F:\zht\code\MobileSAM-master\weights\mobile_sam.pt"
 model_type = "vit_t"
 mobile_sam = sam_model_registry[model_type](checkpoint=sam_checkpoint)

 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 # Load the pre-trained model
+sam_checkpoint = "./mobile_sam.pt"
 model_type = "vit_t"
 mobile_sam = sam_model_registry[model_type](checkpoint=sam_checkpoint)