jianyouli
commited on
Commit
·
fdef182
1
Parent(s):
42403d4
Add application file3
Browse files
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 =
|
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)
|