Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -63,7 +63,7 @@ def yolov8_img_inference(
|
|
63 |
# cls.add(model.names[int(cno)])
|
64 |
|
65 |
# count_result = results.pandas().xyxy[0].value_counts('name')
|
66 |
-
return render, results[0]
|
67 |
|
68 |
# results = model.predict(image, imgsz=image_size, return_outputs=True)
|
69 |
# results = model.predict(image)
|
|
|
63 |
# cls.add(model.names[int(cno)])
|
64 |
|
65 |
# count_result = results.pandas().xyxy[0].value_counts('name')
|
66 |
+
return render, results[0].boxes.cls
|
67 |
|
68 |
# results = model.predict(image, imgsz=image_size, return_outputs=True)
|
69 |
# results = model.predict(image)
|