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(
|