ameerazam08 commited on
Commit
0d0c87b
·
verified ·
1 Parent(s): 73385a0

removed thereshold

Browse files
Files changed (1) hide show
  1. app.py +8 -8
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, conf_threshold):
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
- 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():
 
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():