Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
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))
|