Spaces:

nvidia
/

describe-anything-model-demo

Running on Zero

merve HF Staff commited on Apr 30

Commit

64875b8

verified ·

1 Parent(s): 86f2b4a

Add huggingface-cli download

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,8 +3,7 @@ os.environ["GRADIO_SSR_MODE"] = "false"
 if not os.path.exists("checkpoints"):
     os.makedirs("checkpoints")
-    os.system("pip install gdown")
-    os.system("gdown https://drive.google.com/uc?id=1eQe6blJcyI7oy78C8ozwj1IUkbkFEItf; unzip -o dam_3b_v1.zip -d checkpoints")
 from segment_anything import sam_model_registry, SamPredictor
 import gradio as gr

 if not os.path.exists("checkpoints"):
     os.makedirs("checkpoints")
+    os.system("huggingface-cli download nvidia/DAM-3B --local-dir checkpoints")
 from segment_anything import sam_model_registry, SamPredictor
 import gradio as gr