Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -85,7 +85,8 @@ def yolov8_img_inference(
|
|
85 |
# cls.add(model.names[int(cno)])
|
86 |
|
87 |
# count_result = results.pandas().xyxy[0].value_counts('name')
|
88 |
-
return render, present_objects
|
|
|
89 |
# results = model.predict(image, imgsz=image_size, return_outputs=True)
|
90 |
# results = model.predict(image)
|
91 |
# object_prediction_list = []
|
|
|
85 |
# cls.add(model.names[int(cno)])
|
86 |
|
87 |
# count_result = results.pandas().xyxy[0].value_counts('name')
|
88 |
+
# return render, present_objects
|
89 |
+
return render, {names[k]: v for k, v in present_objects.items()}
|
90 |
# results = model.predict(image, imgsz=image_size, return_outputs=True)
|
91 |
# results = model.predict(image)
|
92 |
# object_prediction_list = []
|