merve HF Staff commited on
Commit
64875b8
·
verified ·
1 Parent(s): 86f2b4a

Add huggingface-cli download

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -3,8 +3,7 @@ os.environ["GRADIO_SSR_MODE"] = "false"
3
 
4
  if not os.path.exists("checkpoints"):
5
  os.makedirs("checkpoints")
6
- os.system("pip install gdown")
7
- os.system("gdown https://drive.google.com/uc?id=1eQe6blJcyI7oy78C8ozwj1IUkbkFEItf; unzip -o dam_3b_v1.zip -d checkpoints")
8
 
9
  from segment_anything import sam_model_registry, SamPredictor
10
  import gradio as gr
 
3
 
4
  if not os.path.exists("checkpoints"):
5
  os.makedirs("checkpoints")
6
+ os.system("huggingface-cli download nvidia/DAM-3B --local-dir checkpoints")
 
7
 
8
  from segment_anything import sam_model_registry, SamPredictor
9
  import gradio as gr