Update web-demos/hugging_face/app.py

web-demos/hugging_face/app.py

@@ -419,9 +419,9 @@ def generate_video_from_frames(frames, output_path, fps=30, bitrate=None):
     #torchvision.io.write_video(output_path, frames, fps=fps, video_codec="libx264")
     #return output_path
     if bitrate is not None:
-        writer = imageio.get_writer(output_path, fps=fps, codec='libx264', bitrate=bitrate, macro_block_size=None,)
+        writer = imageio.get_writer(output_path, fps=fps, codec='libx264rgb', quality=None, output_params=['-crf', '0', '-preset', 'veryslow', '-pix_fmt', 'rgb24']
     else:
-        writer = imageio.get_writer(output_path, fps=fps, codec='libx264', macro_block_size=None)
+        writer = imageio.get_writer(output_path, fps=fps, codec='libx264rgb', quality=None, output_params=['-crf', '0', '-preset', 'veryslow', '-pix_fmt', 'rgb24']
     for frame in frames:
         writer.append_data(frame.astype(np.uint8))
     writer.close()