l

app.py

@@ -42,9 +42,12 @@ def segment(image_pil: Image.Image, prompt: str):
     # Convert PIL image to numpy array
     image_np = np.array(image_pil)
 
+    # Ensure the model is explicitly set to CPU
+    grounder.to("cpu")
+
     # Run GroundingDINO to get boxes for the prompt
     boxes, _, _ = predict(
-        model=grounder.to(device),  # Ensure model is on CPU
+        model=grounder,
         image=image_np,
         caption=prompt,
         box_threshold=0.3,