Update app.py

app.py

@@ -27,7 +27,7 @@ def draw_box(image,det):
         end_point = ((int(ea[2]),int(ea[3])))
         color = (255, 0, 0)
         thickness = 2
-        label = f'{det.data[i]}'
+        label = f'{det.data}'
         font = cv2.FONT_HERSHEY_SIMPLEX  # Choose a font
         font_scale = 1
         color = (0, 0, 255)  # Blue color
@@ -64,7 +64,7 @@ def stream_object_detection(video):
         output_video.write(frame)
         batch = []
         output_video.release()
-        yield output_video_name
+        yield output_video_name,detections
         output_video_name = f"output_{uuid.uuid4()}.mp4"
         output_video = cv2.VideoWriter(output_video_name, video_codec, desired_fps, (width, height)) # type: ignore
         iterating, frame = cap.read()