Update app.py

app.py

@@ -20,14 +20,21 @@ model = YOLO(modin)
 
 def draw_box(image,det):
     height, width, channels = image.shape
-    for ea in det.xyxy:
+    for i,ea in enumerate(det.xyxy):
         #bbox = convert_coords(ea, width, height)
         #print(bbox)
         start_point = ((int(ea[0]),int(ea[1])))
         end_point = ((int(ea[2]),int(ea[3])))
         color = (255, 0, 0)
         thickness = 2
+        label = f'{det.data[i]}
+        font = cv2.FONT_HERSHEY_SIMPLEX  # Choose a font
+        font_scale = 1
+        color = (0, 0, 255)  # Blue color
+        thickness = 2
+        text_position = (int(ea[0]), int(ea[1]) + 10)  # Adjust position as needed
         image = cv2.rectangle(image, start_point, end_point, color, thickness)
+        cv2.putText(image, label, text_position, font, font_scale, color, thickness) 
     return image
 
 @spaces.GPU