freealise commited on
Commit
e05fd91
·
verified ·
1 Parent(s): e66fb8c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -92,12 +92,12 @@ def create_video(frames, fps, type):
92
  clip = ImageSequenceClip(frames, fps=fps)
93
 
94
  maskclip = ImageSequenceClip(frames, fps=fps, ismask=True)
95
- maskclip.to_mask()
96
  clip.set_mask(maskclip)
97
 
98
- clip.write_videofile(type + "_result.mp4", fps=fps)
99
 
100
- return type + "_result.mp4"
101
 
102
 
103
  def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):
 
92
  clip = ImageSequenceClip(frames, fps=fps)
93
 
94
  maskclip = ImageSequenceClip(frames, fps=fps, ismask=True)
95
+ maskclip.to_mask(maskclip, canal=2)
96
  clip.set_mask(maskclip)
97
 
98
+ clip.write_videofile(type + "_result.webm", fps=fps, codec='libvpx9', audio=False)
99
 
100
+ return type + "_result.webm"
101
 
102
 
103
  def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):