Spanicin commited on
Commit
59b079e
·
verified ·
1 Parent(s): 83d2c37

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -179,12 +179,14 @@ def generate_video():
179
  output_video = tempfile.NamedTemporaryFile(dir=TEMP_DIR.name, delete=False, suffix=".mp4").name
180
 
181
  extend_video_simple(temp_video_path, temp_audio_path, output_video)
 
 
182
 
183
  try:
184
  run_inference(
185
- video_path=temp_video_path,
186
  audio_path=temp_audio_path,
187
- video_out_path=output_video,
188
  inference_ckpt_path=inference_ckpt_path,
189
  unet_config_path=unet_config_path,
190
  inference_steps=int(request.form.get('inference_steps', 20)),
 
179
  output_video = tempfile.NamedTemporaryFile(dir=TEMP_DIR.name, delete=False, suffix=".mp4").name
180
 
181
  extend_video_simple(temp_video_path, temp_audio_path, output_video)
182
+ final_output_video = tempfile.NamedTemporaryFile(dir=TEMP_DIR.name, delete=False, suffix="_final_extended.mp4").name
183
+
184
 
185
  try:
186
  run_inference(
187
+ video_path=output_video,
188
  audio_path=temp_audio_path,
189
+ video_out_path=final_output_video,
190
  inference_ckpt_path=inference_ckpt_path,
191
  unet_config_path=unet_config_path,
192
  inference_steps=int(request.form.get('inference_steps', 20)),