diff --git a/src/Python/flask/web/app.py b/src/Python/flask/web/app.py
index 0097c45..1a66899 100644
--- a/src/Python/flask/web/app.py
+++ b/src/Python/flask/web/app.py
@@ -31,14 +31,14 @@ def on_message(client, userdata, message):
         for result in results:
             for box in result.boxes:
                 yolo_results.append({
-                    "class": yolo_classes,
+                    "class": yolo_classes[box.cls.item()],
                     "confidence": box.conf.item(),
                     "bbox": box.xyxy.tolist()
                 })
                 # Draw bounding box on the image
                 x1, y1, x2, y2 = map(int, box.xyxy[0])
                 cv2.rectangle(latest_image, (x1, y1), (x2, y2), (0, 255, 0), 2)
-                cv2.putText(latest_image, f"{yolo_classes} {box.conf.item():.2f}", (x1, y1 - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.9, (0, 255, 0), 2)
+                cv2.putText(latest_image, f"{yolo_classes[box.cls.item()]} {box.conf.item():.2f}", (x1, y1 - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.9, (0, 255, 0), 2)
 
 # Create an MQTT client instance
 mqtt_client = mqtt.Client()