Update app.py

app.py

@@ -60,7 +60,7 @@ def yolov8_img_inference(
     clist= results[0].boxes.cls.tolist()
     cls = set()
     for cno in clist:
-        count = cls.add(model.names)
+        count = cls.add(model.names[int(cno)])
         
     # count_result = results.pandas().xyxy[0].value_counts('name')
     return render, count