PKaushik commited on
Commit
0397646
·
1 Parent(s): d552a82

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -33,10 +33,10 @@ def fn_video(video_file, conf_thres, iou_thres, start_sec, duration):
33
  # Unfortunately, gradio doesn't play too nice with videos rn so we have to do some hackiness
34
  # with ffmpeg at the end of the function here.
35
  with tempfile.NamedTemporaryFile(suffix=".mp4") as temp_file:
36
- out = cv2.VideoWriter(temp_file.name, cv2.VideoWriter_fourcc(*"MP4V"), 60, (1280, 720))
37
 
38
  num_frames = 0
39
- max_frames = duration * 1
40
  while cap.isOpened():
41
  try:
42
  ret, frame = cap.read()
@@ -73,7 +73,7 @@ image_interface = gr.Interface(
73
  title="Human Detection",
74
  description=(
75
  "Gradio demo for Human detection on images. To use it, simply upload your image or click one of the"
76
- " examples to load them. Read more at the links below."
77
  ),
78
  allow_flagging=False,
79
  allow_screenshot=False,
@@ -85,8 +85,8 @@ video_interface = gr.Interface(
85
  gr.Video(type="file"),
86
  gr.Slider(0, 1, value=0.25, label="Confidence Threshold"),
87
  gr.Slider(0, 1, value=0.45, label="IOU Threshold"),
88
- gr.Slider(0, 60, value=0, label="Start Second", step=1),
89
- gr.Slider(0, 60 if pipeline.device.type != 'cpu' else 1, value=2, label="Duration", step=1),
90
  ],
91
  outputs=gr.Video(type="file", format="mp4"),
92
  examples=[
@@ -97,7 +97,7 @@ video_interface = gr.Interface(
97
  title="Human Detection",
98
  description=(
99
  "Gradio demo for Human detection on videos. To use it, simply upload your video or click one of the"
100
- " examples to load them. Read more at the links below."
101
  ),
102
  allow_flagging=False,
103
  allow_screenshot=False,
@@ -115,7 +115,7 @@ webcam_interface = gr.Interface(
115
  title="Human Detection",
116
  description=(
117
  "Gradio demo for Human detection on real time webcam. To use it, simply allow the browser to access"
118
- " your webcam. Read more at the links below."
119
  ),
120
  allow_flagging=False,
121
  allow_screenshot=False,
 
33
  # Unfortunately, gradio doesn't play too nice with videos rn so we have to do some hackiness
34
  # with ffmpeg at the end of the function here.
35
  with tempfile.NamedTemporaryFile(suffix=".mp4") as temp_file:
36
+ out = cv2.VideoWriter(temp_file.name, cv2.VideoWriter_fourcc(*"MP4V"), 120, (1280, 720))
37
 
38
  num_frames = 0
39
+ max_frames = duration * 120
40
  while cap.isOpened():
41
  try:
42
  ret, frame = cap.read()
 
73
  title="Human Detection",
74
  description=(
75
  "Gradio demo for Human detection on images. To use it, simply upload your image or click one of the"
76
+ " examples to load them."
77
  ),
78
  allow_flagging=False,
79
  allow_screenshot=False,
 
85
  gr.Video(type="file"),
86
  gr.Slider(0, 1, value=0.25, label="Confidence Threshold"),
87
  gr.Slider(0, 1, value=0.45, label="IOU Threshold"),
88
+ gr.Slider(0, 120, value=0, label="Start Second", step=1),
89
+ gr.Slider(0, 120 if pipeline.device.type != 'cpu' else 1, value=2, label="Duration", step=1),
90
  ],
91
  outputs=gr.Video(type="file", format="mp4"),
92
  examples=[
 
97
  title="Human Detection",
98
  description=(
99
  "Gradio demo for Human detection on videos. To use it, simply upload your video or click one of the"
100
+ " examples to load them."
101
  ),
102
  allow_flagging=False,
103
  allow_screenshot=False,
 
115
  title="Human Detection",
116
  description=(
117
  "Gradio demo for Human detection on real time webcam. To use it, simply allow the browser to access"
118
+ " your webcam."
119
  ),
120
  allow_flagging=False,
121
  allow_screenshot=False,