Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -29,7 +29,7 @@ for i, url in enumerate(file_urls):
|
|
29 |
f"image_{i}.jpg"
|
30 |
)
|
31 |
|
32 |
-
model = YOLO('
|
33 |
path = [['image_0.jpg'], ['image_1.jpg']]
|
34 |
video_path = [['video.mp4']]
|
35 |
|
@@ -83,11 +83,11 @@ def show_preds_video(video_path):
|
|
83 |
yield cv2.cvtColor(frame_copy, cv2.COLOR_BGR2RGB)
|
84 |
|
85 |
|
86 |
-
|
87 |
gr.components.Video(type="filepath", label="Input Video"),
|
88 |
|
89 |
]
|
90 |
-
|
91 |
gr.components.Image(type="numpy", label="Output Image"),
|
92 |
]
|
93 |
interface_video = gr.Interface(
|
|
|
29 |
f"image_{i}.jpg"
|
30 |
)
|
31 |
|
32 |
+
model = YOLO('best.pt')
|
33 |
path = [['image_0.jpg'], ['image_1.jpg']]
|
34 |
video_path = [['video.mp4']]
|
35 |
|
|
|
83 |
yield cv2.cvtColor(frame_copy, cv2.COLOR_BGR2RGB)
|
84 |
|
85 |
|
86 |
+
inputs_video = [
|
87 |
gr.components.Video(type="filepath", label="Input Video"),
|
88 |
|
89 |
]
|
90 |
+
outputs_video = [
|
91 |
gr.components.Image(type="numpy", label="Output Image"),
|
92 |
]
|
93 |
interface_video = gr.Interface(
|