ford442 commited on
Commit
a597ecf
·
verified ·
1 Parent(s): 05bf5a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=8, 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)
 
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)