Update app.py

app.py

@@ -6,6 +6,18 @@ import gradio as gr
 from segment_anything import SamPredictor, sam_model_registry
 from groundingdino.util.inference import load_model, predict, annotate
 
+checkpoint_path = "./sam_vit_h_4b8939.pth"  # Will be downloaded programmatically
+
+# Download programmatically using hf_hub_download
+from huggingface_hub import hf_hub_download
+
+checkpoint_path = hf_hub_download(
+    repo_id="HCMUE-Research/SAM-vit-h",
+    filename="sam_vit_h_4b8939.pth"
+)
+
+sam = sam_model_registry["vit_h"](checkpoint=checkpoint_path)
+
 grounding_dino_config = "groundingdino/config/GroundingDINO_SwinT_OGC.py"
 grounding_dino_weights = 'groundingdino_swift_ogc.path'