Update app.py

app.py

@@ -1,17 +1,22 @@
 import cv2
 import numpy as np
 import gradio as gr
-import torch
+from huggingface_hub import hf_hub_download
 from segment_anything import sam_model_registry, SamPredictor
+import torch
+
+# Load the Segment Anything Model (SAM) from Hugging Face
+def load_sam_model():
+    checkpoint_path = hf_hub_download(repo_id="facebook/sam-vit-huge", filename="sam_vit_h_4b8939.pth")
+    model_type = "vit_h"
+    device = "cuda" if torch.cuda.is_available() else "cpu"
 
-# Load the Segment Anything Model (SAM)
-sam_checkpoint = "sam_vit_h_4b8939.pth"  # Path to the SAM checkpoint
-model_type = "vit_h"
-device = "cuda" if torch.cuda.is_available() else "cpu"
+    sam = sam_model_registry[model_type](checkpoint=checkpoint_path)
+    sam.to(device=device)
+    predictor = SamPredictor(sam)
+    return predictor
 
-sam = sam_model_registry[model_type](checkpoint=sam_checkpoint)
-sam.to(device=device)
-predictor = SamPredictor(sam)
+predictor = load_sam_model()
 
 def generate_mask(image, event: gr.SelectData):
     """