Spaces:

avanish07
/

yolov8_demo

Runtime error

avanish07 commited on Jul 5, 2023

Commit

bdfd745

•

1 Parent(s): 20087e5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ def show_preds_image(image):
     results = outputs[0].boxes.xyxy.cpu().numpy()
     for det in results:
         x1, y1, x2, y2 = det[:4]
-        label = "Pothole"
         cv2.rectangle(
             image,
             (int(x1), int(y1)),
@@ -51,7 +51,7 @@ interface_image = gr.Interface(
     fn=show_preds_image,
     inputs=inputs_image,
     outputs=outputs_image,
-    title="Pothole detector",
     examples=[],
     cache_examples=False,
 )

     results = outputs[0].boxes.xyxy.cpu().numpy()
     for det in results:
         x1, y1, x2, y2 = det[:4]
+        label = ""
         cv2.rectangle(
             image,
             (int(x1), int(y1)),
     fn=show_preds_image,
     inputs=inputs_image,
     outputs=outputs_image,
+    title="Animal detector",
     examples=[],
     cache_examples=False,
 )