Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,6 @@ from ultralytics import YOLO
|
|
| 8 |
file_urls = [
|
| 9 |
'https://www.dropbox.com/s/b5g97xo901zb3ds/pothole_example.jpg?dl=1',
|
| 10 |
'https://www.dropbox.com/s/86uxlxxlm1iaexa/pothole_screenshot.png?dl=1',
|
| 11 |
-
'https://www.dropbox.com/s/7sjfwncffg8xej2/video_7.mp4?dl=1'
|
| 12 |
]
|
| 13 |
|
| 14 |
def download_file(url, save_name):
|
|
@@ -83,11 +82,11 @@ def show_preds_video(video_path):
|
|
| 83 |
yield cv2.cvtColor(frame_copy, cv2.COLOR_BGR2RGB)
|
| 84 |
|
| 85 |
inputs_video = [
|
| 86 |
-
|
| 87 |
|
| 88 |
]
|
| 89 |
outputs_video = [
|
| 90 |
-
|
| 91 |
]
|
| 92 |
interface_video = gr.Interface(
|
| 93 |
fn=show_preds_video,
|
|
|
|
| 8 |
file_urls = [
|
| 9 |
'https://www.dropbox.com/s/b5g97xo901zb3ds/pothole_example.jpg?dl=1',
|
| 10 |
'https://www.dropbox.com/s/86uxlxxlm1iaexa/pothole_screenshot.png?dl=1',
|
|
|
|
| 11 |
]
|
| 12 |
|
| 13 |
def download_file(url, save_name):
|
|
|
|
| 82 |
yield cv2.cvtColor(frame_copy, cv2.COLOR_BGR2RGB)
|
| 83 |
|
| 84 |
inputs_video = [
|
| 85 |
+
gr.components.Video(type="filepath", label="Input Video"),
|
| 86 |
|
| 87 |
]
|
| 88 |
outputs_video = [
|
| 89 |
+
gr.components.Image(type="numpy", label="Output Image"),
|
| 90 |
]
|
| 91 |
interface_video = gr.Interface(
|
| 92 |
fn=show_preds_video,
|