Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -59,7 +59,7 @@ def yolov8_img_inference(
|
|
59 |
|
60 |
for r in results:
|
61 |
for c in r.boxes.cls:
|
62 |
-
|
63 |
|
64 |
# clist= results[0].boxes.cls.tolist()
|
65 |
# cls = set()
|
@@ -67,7 +67,7 @@ def yolov8_img_inference(
|
|
67 |
# cls.add(model.names[int(cno)])
|
68 |
|
69 |
# count_result = results.pandas().xyxy[0].value_counts('name')
|
70 |
-
return render,
|
71 |
# results = model.predict(image, imgsz=image_size, return_outputs=True)
|
72 |
# results = model.predict(image)
|
73 |
# object_prediction_list = []
|
|
|
59 |
|
60 |
for r in results:
|
61 |
for c in r.boxes.cls:
|
62 |
+
count = print(names[int(c)])
|
63 |
|
64 |
# clist= results[0].boxes.cls.tolist()
|
65 |
# cls = set()
|
|
|
67 |
# cls.add(model.names[int(cno)])
|
68 |
|
69 |
# count_result = results.pandas().xyxy[0].value_counts('name')
|
70 |
+
return render, count
|
71 |
# results = model.predict(image, imgsz=image_size, return_outputs=True)
|
72 |
# results = model.predict(image)
|
73 |
# object_prediction_list = []
|