Update app.py

app.py

@@ -25,13 +25,11 @@ def draw_box(image,det):
         #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}'
-        font = cv2.FONT_HERSHEY_SIMPLEX  # Choose a font
+        label = f'{det.data["class_name"][i]}'
+        font = cv2.FONT_HERSHEY_COMPLEX  # Choose a font
         font_scale = 1
         color = (0, 0, 255)  # Blue color
-        thickness = 2
+        thickness = 1
         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)