Spaces:
Paused
Paused
removed thereshold
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ import os
|
|
7 |
from ultralytics import YOLO
|
8 |
|
9 |
@spaces.GPU()
|
10 |
-
def stream_object_detection(video_path
|
11 |
# Load the YOLO model
|
12 |
model = YOLO("weights/best.pt")
|
13 |
cap = cv2.VideoCapture(video_path)
|
@@ -49,13 +49,13 @@ with gr.Blocks() as app:
|
|
49 |
with gr.Row():
|
50 |
with gr.Column():
|
51 |
video_input = gr.Video(label="Upload Video")
|
52 |
-
conf_threshold = gr.Slider(
|
53 |
-
|
54 |
-
|
55 |
-
|
56 |
-
|
57 |
-
|
58 |
-
)
|
59 |
with gr.Column():
|
60 |
video_output = gr.Video(label="Processed Video")
|
61 |
with gr.Row():
|
|
|
7 |
from ultralytics import YOLO
|
8 |
|
9 |
@spaces.GPU()
|
10 |
+
def stream_object_detection(video_path):
|
11 |
# Load the YOLO model
|
12 |
model = YOLO("weights/best.pt")
|
13 |
cap = cv2.VideoCapture(video_path)
|
|
|
49 |
with gr.Row():
|
50 |
with gr.Column():
|
51 |
video_input = gr.Video(label="Upload Video")
|
52 |
+
# conf_threshold = gr.Slider(
|
53 |
+
# label="Confidence Threshold",
|
54 |
+
# minimum=0.0,
|
55 |
+
# maximum=1.0,
|
56 |
+
# step=0.05,
|
57 |
+
# value=0.30,
|
58 |
+
# )
|
59 |
with gr.Column():
|
60 |
video_output = gr.Video(label="Processed Video")
|
61 |
with gr.Row():
|