Update app.py

app.py

@@ -10,7 +10,7 @@ import urllib.request
 from PIL import Image, ImageDraw
 import matplotlib.pyplot as plt
 
-from Garage.models.GroundedSegmentAnything.segment_anything.segment_anything import SamPredictor, build_sam
+from Garage.models.GroundedSegmentAnything.segment_anything.segment_anything import SamPredictor, build_sam, sam_model_registry
 from Garage.models.GroundedSegmentAnything.GroundingDINO.groundingdino.util.inference import Model
 from Garage import Augmenter
 
@@ -23,7 +23,7 @@ MODEL_DICT = dict(
 
 GROUNDING_DINO_CONFIG_PATH = "Garage/models/GroundedSegmentAnything/GroundingDINO_SwinT_OGC.py"
 GROUNDING_DINO_CHECKPOINT_PATH = "https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth"
-SAM_CHECKPOINT_PATH = "Garage/models/checkpoints/GroundedSegmentAnything/sam_vit_h_4b8939.pth"
+SAM_CHECKPOINT_PATH = "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth"
 SAM_ENCODER_VERSION = "vit_h"
 
 class GradioWindow():