Update app.py

app.py

@@ -55,10 +55,14 @@ def yolov8_img_inference(
     # obj_id = list(names)[list(names.values()).index([0])]
     # count 'car' objects in the results
     # count_result = results[0].boxes.cls[0].item()
-    count_result = results[0].boxes.cls.tolist().pandas()
+    # count_result = results[0].boxes.cls.tolist()
+    clist= res[0].boxes.cls
+    cls = set()
+    for cno in clist:
+        cls.add(model.names[int(cno)])
   
     # count_result = results.pandas().xyxy[0].value_counts('name')
-    return render, count_result 
+    return render, cls 
     
     # results = model.predict(image, imgsz=image_size, return_outputs=True)
     # results = model.predict(image)