Update app.py

app.py

@@ -1,5 +1,4 @@
 import os
-import urllib
 
 import cv2
 import gradio as gr
@@ -11,17 +10,8 @@ from PIL import Image
 from segment_anything import SamAutomaticMaskGenerator, sam_model_registry
 
 
-# download model weights
-ckpts_dir = os.path.join(os.getcwd() + "/ckpts")
-if not os.path.exists(ckpts_dir):
-    os.makedirs(ckpts_dir)
-ckpt_path = os.path.join(ckpts_dir + "/sam_vit_b_01ec64.pth")
-if not os.path.exists(ckpt_path):
-    url = "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth"
-    urllib.request.urlretrieve(url, filename=ckpt_path)
-
 # setup model
-sam = sam_model_registry["vit_b"](checkpoint=ckpt_path)
+sam = sam_model_registry["vit_b"](checkpoint="./sam_vit_b_01ec64.pth")
 mask_generator = SamAutomaticMaskGenerator(sam)
 
 # copied from: https://github.com/facebookresearch/segment-anything