Update app.py
Browse files
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
|
| 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 |
|