Spaces:

prodevroger
/

gg

Sleeping

IZERE HIRWA Roger commited on Jul 11

Commit

3ec0a4b

1 Parent(s): 5ee91ca

1

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,7 +44,7 @@ def segment(image_pil: Image.Image, prompt: str):
     # Run GroundingDINO to get boxes for the prompt
     boxes, _, _ = predict(
-        model=grounder,
         image=image_np,
         caption=prompt,
         box_threshold=0.3,

     # Run GroundingDINO to get boxes for the prompt
     boxes, _, _ = predict(
+        model=grounder.to(device),  # Ensure model is on CPU
         image=image_np,
         caption=prompt,
         box_threshold=0.3,