Upload app.py

app.py

@@ -1,11 +1,6 @@
 import os
 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
 import numpy as np
@@ -96,7 +91,7 @@ def describe_without_streaming(image_base64: str, mask_base64: str, query: str):
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(description="Describe Anything gradio demo")
-    parser.add_argument("--model-path", type=str, default="checkpoints/dam_3b_v1", help="Path to the model checkpoint")
+    parser.add_argument("--model-path", type=str, default="nvidia/DAM-3B", help="Path to the model checkpoint")
     parser.add_argument("--prompt-mode", type=str, default="full+focal_crop", help="Prompt mode")
     parser.add_argument("--conv-mode", type=str, default="v1", help="Conversation mode")
     parser.add_argument("--temperature", type=float, default=0.2, help="Sampling temperature")