Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -197,7 +197,7 @@ def generate(prompt, negative_prompt, clips_list, input_image_filepath, input_vi
|
|
| 197 |
result_images_tensor = result_images_tensor[:, :, :actual_num_frames, pad_t:(-pad_b or None), pad_l:(-pad_r or None)]
|
| 198 |
video_np = (np.clip(result_images_tensor[0].permute(1, 2, 3, 0).cpu().float().numpy(), 0, 1) * 255).astype(np.uint8)
|
| 199 |
output_video_path = os.path.join(tempfile.mkdtemp(), f"output_{random.randint(10000,99999)}.mp4")
|
| 200 |
-
with imageio.get_writer(output_video_path, format='FFMPEG', fps=call_kwargs["frame_rate"], codec='libx264', quality=
|
| 201 |
for idx, frame in enumerate(video_np):
|
| 202 |
progress(idx / len(video_np), desc="Saving video clip...")
|
| 203 |
video_writer.append_data(frame)
|
|
|
|
| 197 |
result_images_tensor = result_images_tensor[:, :, :actual_num_frames, pad_t:(-pad_b or None), pad_l:(-pad_r or None)]
|
| 198 |
video_np = (np.clip(result_images_tensor[0].permute(1, 2, 3, 0).cpu().float().numpy(), 0, 1) * 255).astype(np.uint8)
|
| 199 |
output_video_path = os.path.join(tempfile.mkdtemp(), f"output_{random.randint(10000,99999)}.mp4")
|
| 200 |
+
with imageio.get_writer(output_video_path, format='FFMPEG', fps=call_kwargs["frame_rate"], codec='libx264', quality=10, pixelformat='yuv420p') as video_writer:
|
| 201 |
for idx, frame in enumerate(video_np):
|
| 202 |
progress(idx / len(video_np), desc="Saving video clip...")
|
| 203 |
video_writer.append_data(frame)
|