Spaces:

openfoodfacts
/

crop-detection

Sleeping

jeremyarancio commited on Sep 27

Commit

cebced8

•

1 Parent(s): 1e31676

Init

Files changed (2) hide show

app.py CHANGED Viewed

@@ -24,8 +24,11 @@ def detect_and_crop(image: Image.Image) -> Image.Image:
     results = MODEL.predict(image,**INF_PARAMETERS)
     result = results[0]
     for box in result.boxes.xyxy.cpu().numpy():
-        cropped_image = image.crop(box=box)
-    return cropped_image
 # Gradio UI

     results = MODEL.predict(image,**INF_PARAMETERS)
     result = results[0]
     for box in result.boxes.xyxy.cpu().numpy():
+        if len(box) > 0:
+            cropped_image = image.crop(box=box)
+            return cropped_image
+        else:
+            return image
 # Gradio UI

images/000_067_500_6567_1.jpg ADDED Viewed