Update app.py

app.py

@@ -14,7 +14,7 @@ print(f"    > Device selected: {'GPU' if device == 0 else 'CPU'}")
 
 # Step 2: Load SAM Model
 print("Step 2: Loading SAM Model...")
-generator = pipeline("mask-generation", model="syscv-community/sam-hq-vit-huge", device=device)
+generator = pipeline("mask-generation", model="facebook/sam-vit-huge", device=device)
 print("    > SAM Model loaded successfully.")
 
 
@@ -30,7 +30,7 @@ def segment_image(image):
     print(f"    > Original size: {original_size}, Resized size: {resized_size}")
 
     # Run SAM Segmentation
-    print("    > Running SAM-HQ segmentation...")
+    print("    > Running SAM segmentation...")
     outputs = generator(raw_image, points_per_batch=64)
     masks = outputs["masks"]
     print(f"    > {len(masks)} masks generated.")