xinghaochen commited on
Commit
8cfa025
1 Parent(s): 7d0a798

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -12,6 +12,7 @@ from torchvision.transforms import ToTensor
12
  from utils.tools import format_results, point_prompt
13
  from utils.tools_gradio import fast_process
14
  from tinysam import sam_model_registry, SamPredictor
 
15
 
16
 
17
  snapshot_download("merve/tinysam", local_dir="tinysam")
 
12
  from utils.tools import format_results, point_prompt
13
  from utils.tools_gradio import fast_process
14
  from tinysam import sam_model_registry, SamPredictor
15
+ from huggingface_hub import snapshot_download
16
 
17
 
18
  snapshot_download("merve/tinysam", local_dir="tinysam")