Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -22,7 +22,8 @@ def yolov8_inference(
|
|
22 |
model = YOLO("ultralyticsplus/yolov8s")
|
23 |
model.conf = conf_threshold
|
24 |
model.iou = iou_threshold
|
25 |
-
results = model.predict(image, return_outputs=True)
|
|
|
26 |
object_prediction_list = []
|
27 |
annotator = Annotator(image)
|
28 |
for _, image_results in enumerate(results):
|
|
|
22 |
model = YOLO("ultralyticsplus/yolov8s")
|
23 |
model.conf = conf_threshold
|
24 |
model.iou = iou_threshold
|
25 |
+
# results = model.predict(image, return_outputs=True)
|
26 |
+
results = model.predict(image)
|
27 |
object_prediction_list = []
|
28 |
annotator = Annotator(image)
|
29 |
for _, image_results in enumerate(results):
|