Update app.py

app.py

@@ -64,7 +64,7 @@ def yolov8_img_inference(
     #     for c in r.boxes.cls:
     #         count = names[int(c)]
     # count_result = results.pandas().xyxy[0].value_counts('name')
-    return render, count 
+    return render, cls
     
     # results = model.predict(image, imgsz=image_size, return_outputs=True)
     # results = model.predict(image)