Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ def detect_objects(image):
|
|
19 |
for box,cls in zip(boxes,classes):
|
20 |
annotator.box_label(box, label=names[int(cls)], color=colors(int(cls)))
|
21 |
annotator1.box_label(box, label=None, color=colors(int(cls)))
|
22 |
-
return Image.fromarray(annotator.result()),Image.fromarray(annotator1.result())
|
23 |
|
24 |
# Gradio Interface
|
25 |
title = "YOLOv8 Object Detection"
|
|
|
19 |
for box,cls in zip(boxes,classes):
|
20 |
annotator.box_label(box, label=names[int(cls)], color=colors(int(cls)))
|
21 |
annotator1.box_label(box, label=None, color=colors(int(cls)))
|
22 |
+
return Image.fromarray(annotator.result()),Image.fromarray(annotator1.result()),None
|
23 |
|
24 |
# Gradio Interface
|
25 |
title = "YOLOv8 Object Detection"
|