Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -4,6 +4,7 @@ from detect_deepsort import run_deepsort
|
|
4 |
from detect_strongsort import run_strongsort
|
5 |
from detect import run
|
6 |
import os
|
|
|
7 |
from PIL import Image
|
8 |
import cv2
|
9 |
import numpy as np
|
@@ -110,9 +111,9 @@ def app():
|
|
110 |
],
|
111 |
value="None"
|
112 |
)
|
|
|
113 |
gr.Examples(['./img_examples/Exam_1.png','./img_examples/Exam_2.png','./img_examples/Exam_3.png','./img_examples/Exam_4.png','./img_examples/Exam_5.png'], inputs=img_path,label = "Image Example", cache_examples = False)
|
114 |
gr.Examples(['./video_examples/video_1.mp4', './video_examples/video_2.mp4','./video_examples/video_3.mp4','./video_examples/video_4.mp4','./video_examples/video_5.mp4'], inputs=vid_path, label = "Video Example", cache_examples = False)
|
115 |
-
yolov9_infer = gr.Button(value="Inference")
|
116 |
with gr.Column():
|
117 |
gr.HTML("<h2>Output</h2>")
|
118 |
output_image = gr.Image(type="numpy",label="Output")
|
|
|
4 |
from detect_strongsort import run_strongsort
|
5 |
from detect import run
|
6 |
import os
|
7 |
+
import torch
|
8 |
from PIL import Image
|
9 |
import cv2
|
10 |
import numpy as np
|
|
|
111 |
],
|
112 |
value="None"
|
113 |
)
|
114 |
+
yolov9_infer = gr.Button(value="Inference")
|
115 |
gr.Examples(['./img_examples/Exam_1.png','./img_examples/Exam_2.png','./img_examples/Exam_3.png','./img_examples/Exam_4.png','./img_examples/Exam_5.png'], inputs=img_path,label = "Image Example", cache_examples = False)
|
116 |
gr.Examples(['./video_examples/video_1.mp4', './video_examples/video_2.mp4','./video_examples/video_3.mp4','./video_examples/video_4.mp4','./video_examples/video_5.mp4'], inputs=vid_path, label = "Video Example", cache_examples = False)
|
|
|
117 |
with gr.Column():
|
118 |
gr.HTML("<h2>Output</h2>")
|
119 |
output_image = gr.Image(type="numpy",label="Output")
|