Spaces:
Running
on
Zero
Running
on
Zero
Add huggingface-cli download
Browse files
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("
|
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
|