Fix display

.gitignore

@@ -1 +1,2 @@
-**/.DS_Store
+**/.DS_Store
+gradio_cached_examples

app.py

@@ -61,13 +61,14 @@ def predict_image(image, threshold):
     # drow boxes on image
     draw = ImageDraw.Draw(image)
 
-    for (class_name, coords, confidence) in detections:
+    for detection in detections:
+        coords = detection['xyxyxyxy']
         if len(coords) != 4:
-            raise ValueError("Each detection should be a polygon with 8 coordinates (xyxyxyxy).")
+            raise ValueError("Each detection should be a polygon with 4 coordinates (xyxyxyxy).")
 
         points = [(coord[0], coord[1]) for coord in coords]
         draw.polygon(points, outline="red", width=LINE_WIDTH)
-        draw.text((points[0][0], points[0][1]), class_name, fill="red")
+        draw.text((points[0][0], points[0][1]), detection['class_name'], fill="red")
 
     return image, img.shape, len(detections), duration