Ivan Felipe Rodriguez commited on
Commit
43e6405
·
1 Parent(s): d977e82

class num was failing

Browse files
Files changed (2) hide show
  1. app.py +4 -4
  2. inference.py +1 -1
app.py CHANGED
@@ -205,7 +205,7 @@ with gr.Blocks(title='Shark Patrol',theme=gr.themes.Soft(),live=True,) as demo:
205
 
206
  with gr.Tab("Test your own Video"):
207
  with gr.Row():
208
- video_input = gr.Video(source='upload',include_audio=False)
209
  #video_input.style(witdh='50%',height='50%')
210
  video_output = gr.Video()
211
  #video_output.style(witdh='50%',height='50%')
@@ -217,11 +217,11 @@ with gr.Blocks(title='Shark Patrol',theme=gr.themes.Soft(),live=True,) as demo:
217
  example_images = gr.Dataset(components=[video_input],
218
  samples=[[path.as_posix()]
219
  for path in paths])
220
- example_images.click(fn=set_example_image,
 
 
221
  inputs=example_images,
222
  outputs=video_input)
223
- video_button.click(analyze_video_parallel, inputs=video_input, outputs=video_output)
224
-
225
 
226
 
227
  demo.queue()
 
205
 
206
  with gr.Tab("Test your own Video"):
207
  with gr.Row():
208
+ video_input = gr.Video(source='upload')
209
  #video_input.style(witdh='50%',height='50%')
210
  video_output = gr.Video()
211
  #video_output.style(witdh='50%',height='50%')
 
217
  example_images = gr.Dataset(components=[video_input],
218
  samples=[[path.as_posix()]
219
  for path in paths])
220
+
221
+ video_button.click(analyze_video_parallel, inputs=video_input, outputs=video_output)
222
+ example_images.click(fn=set_example_image,
223
  inputs=example_images,
224
  outputs=video_input)
 
 
225
 
226
 
227
  demo.queue()
inference.py CHANGED
@@ -74,7 +74,7 @@ classes= [ 'Beach',
74
  'Tiger shark',
75
  'Bull shark']
76
 
77
- classes +=['None']*50
78
 
79
 
80
 
 
74
  'Tiger shark',
75
  'Bull shark']
76
 
77
+ classes +=['None']*100
78
 
79
 
80