nagasurendra commited on
Commit
efee8a7
·
verified ·
1 Parent(s): 33ab799

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -30,8 +30,8 @@ def process_video(video):
30
  # Perform inference on the frame
31
  results = model(frame)
32
 
33
- # Extract annotated image from results
34
- annotated_frame = results.render()[0] # This will give the frame with bounding boxes
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)