jianyouli commited on
Commit
fdef182
1 Parent(s): 42403d4

Add application file3

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ from utils.tools_gradio import fast_process
13
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
14
 
15
  # Load the pre-trained model
16
- sam_checkpoint = r"F:\zht\code\MobileSAM-master\weights\mobile_sam.pt"
17
  model_type = "vit_t"
18
 
19
  mobile_sam = sam_model_registry[model_type](checkpoint=sam_checkpoint)
 
13
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
14
 
15
  # Load the pre-trained model
16
+ sam_checkpoint = "./mobile_sam.pt"
17
  model_type = "vit_t"
18
 
19
  mobile_sam = sam_model_registry[model_type](checkpoint=sam_checkpoint)