Update app.py
Browse files
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=
|
| 186 |
audio_path=temp_audio_path,
|
| 187 |
-
video_out_path=
|
| 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)),
|