piperod91 commited on
Commit
b1c2f87
1 Parent(s): a2d4eea

changing video input for file input

Browse files
Files changed (1) hide show
  1. app.py +13 -4
app.py CHANGED
@@ -174,7 +174,13 @@ def draw_cockpit(frame, top_pred,cnt):
174
 
175
  def process_video(input_video, out_fps = 'auto', skip_frames = 7):
176
  print('Processing video: ')
177
- cap = cv2.VideoCapture(input_video)
 
 
 
 
 
 
178
 
179
  output_path = "output.mp4"
180
  if out_fps != 'auto' and type(out_fps) == int:
@@ -261,13 +267,16 @@ def process_video(input_video, out_fps = 'auto', skip_frames = 7):
261
  yield None, output_path
262
 
263
  with gr.Blocks(theme=theme) as demo:
 
 
 
264
  with gr.Row().style(equal_height=True,height='25%'):
265
- input_video = gr.Video(label="Input")
266
  original_frames = gr.Image(label="Processed Frame").style( height=650)
267
  #processed_frames = gr.Image(label="Shark Engine")
268
- output_video = gr.Video(label="Output Video")
269
- #dashboard = gr.Image(label="Events")
270
 
 
 
271
  with gr.Row():
272
  paths = sorted(pathlib.Path('videos_example/').rglob('*.mp4'))
273
  samples=[[path.as_posix()] for path in paths if 'raw_videos' in str(path)]
 
174
 
175
  def process_video(input_video, out_fps = 'auto', skip_frames = 7):
176
  print('Processing video: ')
177
+
178
+ try:
179
+ cap = cv2.VideoCapture(input_video.name)
180
+ except:
181
+ cap = cv2.VideoCapture(input_video)
182
+
183
+
184
 
185
  output_path = "output.mp4"
186
  if out_fps != 'auto' and type(out_fps) == int:
 
267
  yield None, output_path
268
 
269
  with gr.Blocks(theme=theme) as demo:
270
+ input_video = gr.File(label="Input")
271
+
272
+
273
  with gr.Row().style(equal_height=True,height='25%'):
274
+
275
  original_frames = gr.Image(label="Processed Frame").style( height=650)
276
  #processed_frames = gr.Image(label="Shark Engine")
 
 
277
 
278
+ #dashboard = gr.Image(label="Events")
279
+ output_video = gr.File(label="Output Video")
280
  with gr.Row():
281
  paths = sorted(pathlib.Path('videos_example/').rglob('*.mp4'))
282
  samples=[[path.as_posix()] for path in paths if 'raw_videos' in str(path)]