Add application file

app.py

@@ -5,12 +5,24 @@ import cv2
 from segment_anything import sam_model_registry, SamAutomaticMaskGenerator
 from PIL import Image
 
-# Model yükle
-sam_checkpoint = "sam_vit_b.pth"
+
+import os
+import urllib.request
+
+MODEL_URL = "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth"
+MODEL_PATH = "sam_vit_b.pth"
+
+# Eğer model yoksa indir
+if not os.path.exists(MODEL_PATH):
+    print("Model indiriliyor...")
+    urllib.request.urlretrieve(MODEL_URL, MODEL_PATH)
+    print("Model indirildi.")
+
+# Model yükle 
 model_type = "vit_b"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-sam = sam_model_registry[model_type](checkpoint=sam_checkpoint)
+sam = sam_model_registry[model_type](checkpoint=MODEL_PATH)
 sam.to(device=device)
 
 mask_generator = SamAutomaticMaskGenerator(sam)