Update app.py
Browse files
app.py
CHANGED
@@ -9,8 +9,7 @@ from huggingface_hub import hf_hub_download
|
|
9 |
from segment_anything import SamPredictor, sam_model_registry
|
10 |
|
11 |
|
12 |
-
hf_hub_download(repo_id="vmoras/sam_api", token=
|
13 |
-
|
14 |
|
15 |
sam_checkpoint = "sam_vit_h.pth"
|
16 |
model_type = "vit_h"
|
|
|
9 |
from segment_anything import SamPredictor, sam_model_registry
|
10 |
|
11 |
|
12 |
+
hf_hub_download(repo_id="vmoras/sam_api", filename="sam_vit_h.pth", token=os.environ.get('model_token'), local_dir="./")
|
|
|
13 |
|
14 |
sam_checkpoint = "sam_vit_h.pth"
|
15 |
model_type = "vit_h"
|