Update app.py

app.py

@@ -10,9 +10,9 @@ from huggingface_hub import hf_hub_download
 from segment_anything import SamPredictor, sam_model_registry
 
 
-#hf_hub_download(repo_id="vmoras/sam_api", filename="sam_vit_h.pth", token=os.environ.get('model_token'), local_dir="./") 
+hf_hub_download(repo_id="vmoras/sam_api", filename="sam_vit_h.pth", token=os.environ.get('model_token'), local_dir="./") 
 
-sam_checkpoint = "sam_vit_h_0.pth"
+sam_checkpoint = "sam_vit_h.pth"
 model_type = "vit_h"
 device = 'cuda' if torch.cuda.is_available() else 'cpu'