Update web-demos/hugging_face/app.py

web-demos/hugging_face/app.py

@@ -416,9 +416,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)
+        writer = imageio.get_writer(output_path, fps=fps, codec='libx264', bitrate=bitrate, macro_block_size=None,)
     else:
-        writer = imageio.get_writer(output_path, fps=fps, codec='libx264')
+        writer = imageio.get_writer(output_path, fps=fps, codec='libx264', macro_block_size=None)
     for frame in frames:
         writer.append_data(frame.astype(np.uint8))
     writer.close()