Update app.py

app.py

@@ -15,7 +15,7 @@ import gradio as gr
 from huggingface_hub import snapshot_download
 
 from image_tools.sizes import resize_and_crop
-from moviepy.editor import *
+#from moviepy.editor import *
 
 
 model = snapshot_download(repo_id="akhaliq/frame-interpolation-film-style")
@@ -87,10 +87,10 @@ def get_frames(video_in, step, name, resize_w):
 
 def create_video(frames, fps, type):
     print("building video result")
-    for j, img in enumerate(frames):
-        frames[j] = cv2.cvtColor(cv2.imread(img).astype(np.uint8), cv2.COLOR_RGB2RGBA)
+    #for j, img in enumerate(frames):
+    #    frames[j] = cv2.cvtColor(cv2.imread(img).astype(np.uint8), cv2.COLOR_RGB2RGBA)
 
-    mediapy.write_video(type + "_result.webm", frames, fps=fps, codec='vp9', encoded_format='rgba')
+    mediapy.write_video(type + "_result.mp4", frames, fps=fps)
     
     #https://zulko.github.io/moviepy/getting_started/videoclips.html#mask-clips
     #clip = ImageSequenceClip(frames, fps=fps)
@@ -99,7 +99,7 @@ def create_video(frames, fps, type):
     #clip.set_mask(maskclip)
     #clip.write_videofile(type + "_result.mp4", fps=fps, codec='libx265', ffmpeg_params=['-c:v', 'libx265', '-x265-params', 'crf=25', '-pix_fmt', 'yuv420p10le', '-an'], audio=False)
     
-    return type + "_result.webm"
+    return type + "_result.mp4"
 
     
 def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):