fffiloni commited on
Commit
0d7b1f5
·
verified ·
1 Parent(s): 0683253

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -391,6 +391,12 @@ def propagate_to_all(video_in, checkpoint, stored_inference_state, stored_frame_
391
  mask_clip = ImageSequenceClip(masks_frames, fps=fps)
392
  # Write the result to a file
393
  mask_final_vid_output_path = "mask_output_video.mp4"
 
 
 
 
 
 
394
 
395
  return gr.update(value=None), gr.update(value=final_vid_output_path), working_frame, available_frames_to_check, gr.update(visible=True), mask_final_vid_output_path
396
 
 
391
  mask_clip = ImageSequenceClip(masks_frames, fps=fps)
392
  # Write the result to a file
393
  mask_final_vid_output_path = "mask_output_video.mp4"
394
+
395
+ # Write the result to a file
396
+ mask_clip.write_videofile(
397
+ mask_final_vid_output_path,
398
+ codec='libx264'
399
+ )
400
 
401
  return gr.update(value=None), gr.update(value=final_vid_output_path), working_frame, available_frames_to_check, gr.update(visible=True), mask_final_vid_output_path
402