Update web-demos/hugging_face/app.py

web-demos/hugging_face/app.py

@@ -334,7 +334,7 @@ def inpaint_video(video_state, resize_ratio_number, dilate_radius_number, raft_i
 
 
 # generate video after vos inference
-def generate_video_from_frames(frames, output_path, fps=30):
+def generate_video_from_frames(frames, output_path, fps=30, bitrate=None):
     """
     Generates a video from a list of frames.
     
@@ -369,7 +369,10 @@ def generate_video_from_frames(frames, output_path, fps=30):
     # Write the video
     #torchvision.io.write_video(output_path, frames, fps=fps, video_codec="libx264")
     #return output_path
-    writer = imageio.get_writer(output_path, fps=fps, codec='libx264', bitrate=bitrate)
+    if bitrate is not None:
+        writer = imageio.get_writer(output_path, fps=fps, codec='libx264', bitrate=bitrate)
+    else:
+        writer = imageio.get_writer(output_path, fps=fps, codec='libx264')
     for frame in frames:
         writer.append_data(frame.astype(np.uint8))
     writer.close()