u

app.py

@@ -66,8 +66,9 @@ def segment(image_pil: Image.Image, prompt: str):
 
     # 2) Largest box → mask via SAM
     box = boxes[np.argmax((boxes[:,2]-boxes[:,0])*(boxes[:,3]-boxes[:,1]))]
+    box_np = box.cpu().numpy() if hasattr(box, "cpu") else np.array(box)
     predictor.set_image(np.array(image_pil))
-    masks, _, _ = predictor.predict(box=box)
+    masks, _, _ = predictor.predict(box=box_np)
     mask = masks[0]  # boolean HxW
 
     return mask