Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -30,8 +30,8 @@ def process_video(video):
|
|
30 |
# Perform inference on the frame
|
31 |
results = model(frame)
|
32 |
|
33 |
-
#
|
34 |
-
annotated_frame = results.
|
35 |
|
36 |
# Write the annotated frame to the output video
|
37 |
output_video.write(annotated_frame)
|
|
|
30 |
# Perform inference on the frame
|
31 |
results = model(frame)
|
32 |
|
33 |
+
# The results object contains annotations for the frame
|
34 |
+
annotated_frame = results[0].plot() # Plot the frame with bounding boxes
|
35 |
|
36 |
# Write the annotated frame to the output video
|
37 |
output_video.write(annotated_frame)
|