ibyteohdear commited on
Commit
2637b7e
·
verified ·
1 Parent(s): b346449

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -14
app.py CHANGED
@@ -170,22 +170,9 @@ with gr.Blocks() as demo:
170
  v_seed = gr.Number(label="Seed (-1 = random)", value=-1)
171
  v_steps = gr.Slider(label="Num Steps", minimum=1, maximum=100, value=25, step=1)
172
  v_cfg = gr.Slider(label="Guidance Strength", minimum=1, maximum=15, value=4.5, step=0.1)
173
- v_dur = gr.Number(label="Detected Duration (sec)", value=8)
174
 
175
  v_btn = gr.Button("Generate Video Audio")
176
  v_out = gr.Video(label="Generated Video with Audio")
177
-
178
- def get_video_duration(video_path):
179
- if video_path is None:
180
- return 8
181
- try:
182
- info = torchaudio.info(video_path)
183
-
184
- duration = info.num_frames / info.sample_rate
185
- return round(duration, 2)
186
- except Exception:
187
- return 8
188
-
189
  video_in.change(
190
  fn=get_video_duration,
191
  inputs=video_in,
@@ -194,7 +181,7 @@ with gr.Blocks() as demo:
194
 
195
  v_btn.click(
196
  fn=video_to_audio,
197
- inputs=[video_in, v_prompt, v_neg, v_seed, v_steps, v_cfg, v_dur],
198
  outputs=v_out
199
  )
200
 
 
170
  v_seed = gr.Number(label="Seed (-1 = random)", value=-1)
171
  v_steps = gr.Slider(label="Num Steps", minimum=1, maximum=100, value=25, step=1)
172
  v_cfg = gr.Slider(label="Guidance Strength", minimum=1, maximum=15, value=4.5, step=0.1)
 
173
 
174
  v_btn = gr.Button("Generate Video Audio")
175
  v_out = gr.Video(label="Generated Video with Audio")
 
 
 
 
 
 
 
 
 
 
 
 
176
  video_in.change(
177
  fn=get_video_duration,
178
  inputs=video_in,
 
181
 
182
  v_btn.click(
183
  fn=video_to_audio,
184
+ inputs=[video_in, v_prompt, v_neg, v_seed, v_steps, v_cfg],
185
  outputs=v_out
186
  )
187