Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 os.system("git clone https://github.com/google-research/frame-interpolation")
 import sys
 sys.path.append("frame-interpolation")
+import cv2
 import numpy as np
 import tensorflow as tf
 import mediapy
@@ -70,14 +71,48 @@ def predict(frame1, frame2, times_to_interpolate):
         util.interpolate_recursively_from_files(
             input_frames, times_to_interpolate, interpolator))
     print(frames)  
-    #mediapy.write_video("out.mp4", frames, fps=30)
-    #return "out.mp4"
-    mediapy.to_uint8(frames)
-    return frames
+    mediapy.write_video("out.mp4", frames, fps=30)
+
+    # video.mp4 is a video of 9 seconds
+        filename = "out.mp4"
+        
+        cap = cv2.VideoCapture(filename)
+        cap.set(cv2.CAP_PROP_POS_AVI_RATIO,0)
+        frameCount = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
+        frameWidth = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+        frameHeight = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
+        videoFPS = int(cap.get(cv2.CAP_PROP_FPS))
+        
+        print (f"frameCount: {frameCount}")
+        print (f"frameWidth: {frameWidth}")
+        print (f"frameHeight: {frameHeight}")
+        print (f"videoFPS: {videoFPS}")
+        
+        buf = np.empty((
+            frameCount,
+            frameHeight,
+            frameWidth,
+            3), np.dtype('uint8'))
+        
+        fc = 0
+        ret = True
+        
+        while (fc < frameCount):
+            ret, buf[fc] = cap.read()
+            fc += 1
+        
+        cap.release()
+        videoArray = buf
+        
+        print (f"DURATION: {frameCount/videoFPS}")
+    
+    return "out.mp4"
+    
+    
     
 title="sketch-frame-interpolation"
 description="This is a fork of the Gradio demo for FILM: Frame Interpolation for Large Scene Motion from @akhaliq, but using sketches instead of images. This could be very useful for the animation industry :) <br /> To use it, simply draw your sketches and add the times to interpolate number. Read more at the links below."
 article = "<p style='text-align: center'><a href='https://film-net.github.io/' target='_blank'>FILM: Frame Interpolation for Large Motion</a> | <a href='https://github.com/google-research/frame-interpolation' target='_blank'>Github Repo</a></p>"
 custom_css = "style.css"
 
-gr.Interface(predict,[sketch1,sketch2,slider],outputs=gr.outputs.Carousel([gr.outputs.Image(type="list")]),title=title,description=description,article=article, css=custom_css).launch(enable_queue=True)
+gr.Interface(predict,[sketch1,sketch2,slider],'playable_video',title=title,description=description,article=article, css=custom_css).launch(enable_queue=True)