vmoras commited on
Commit
1a25549
1 Parent(s): 00b626e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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= os.environ.get('model_token'), filename="sam_vit_h.pth")
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"