nagasurendra commited on
Commit
e140e6c
·
verified ·
1 Parent(s): 4dd8158

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,8 +14,8 @@ model_yolo11 = safe_load_yolo_model('./data/yolo11n.pt')
14
  model_best = safe_load_yolo_model('./data/best.pt')
15
 
16
  def process_video(video):
17
- # Read video input
18
- cap = cv2.VideoCapture(video.name)
19
  fps = cap.get(cv2.CAP_PROP_FPS)
20
  frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
21
  frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
 
14
  model_best = safe_load_yolo_model('./data/best.pt')
15
 
16
  def process_video(video):
17
+ # Read video input (Gradio gives you a file-like object)
18
+ cap = cv2.VideoCapture(video)
19
  fps = cap.get(cv2.CAP_PROP_FPS)
20
  frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
21
  frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))