Update app.py

app.py

@@ -38,7 +38,7 @@ def do_interpolation(frame1, frame2, interpolation):
     mediapy.write_video(f"{frame1}_to_{frame2}_out.mp4", frames, fps=25)
     return f"{frame1}_to_{frame2}_out.mp4"
     
-def get_frames(video_in, step, name):
+def get_frames(video_in, step, name, resize):
     frames = []
     cap = cv2.VideoCapture(video_in)
     cframes = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
@@ -70,6 +70,7 @@ def get_frames(video_in, step, name):
         ret, frame = cap.read()
         if ret == False:
             break
+        frame = cv2.resize(frame, (int(resize), int(resize/2))) 
         cv2.imwrite(f"{name}_{step}{str(i)}.jpg",frame)
         frames.append(f"{name}_{step}{str(i)}.jpg")
         i+=1
@@ -89,11 +90,11 @@ def create_video(frames, fps, type):
     return type + "_result.mp4"
 
     
-def infer(url_in,interpolation,fps_output):
+def infer(url_in,interpolation,resize,fps_output):
     
     fps_output = logscale(fps_output)
     # 1. break video into frames and get FPS
-    break_vid = get_frames(url_in, "vid_input_frame", "origin")
+    break_vid = get_frames(url_in, "vid_input_frame", "origin", resize)
     frames_list= break_vid[0]
     fps = break_vid[1]
     print(f"ORIGIN FPS: {fps}")
@@ -191,6 +192,7 @@ with gr.Blocks() as demo:
         with gr.Row():
             with gr.Column():
                 url_input = gr.Textbox(value="./examples/streetview.mp4", label="URL")
+                resize_num = gr.Number(value=256, label="Resize to width: ")
                 with gr.Row():
                     interpolation_slider = gr.Slider(minimum=1, maximum=5, step=1, value=1, label="Interpolation Steps: ")
                     interpolation = gr.Label(value=2, show_label=False)
@@ -209,11 +211,11 @@ with gr.Blocks() as demo:
     gr.Examples(
         examples=[["./examples/streetview.mp4", 1, 1]],
         fn=infer,
-        inputs=[url_input,interpolation_slider,fps_output_slider],
+        inputs=[url_input,interpolation_slider,resize_num,fps_output_slider],
         outputs=[video_output,file_output,depth_output],
         cache_examples=True
     )
     
-    submit_btn.click(fn=infer, inputs=[url_input,interpolation_slider,fps_output_slider], outputs=[video_output, file_output, depth_output])
+    submit_btn.click(fn=infer, inputs=[url_input,interpolation_slider,resize_num,fps_output_slider], outputs=[video_output, file_output, depth_output])
 
 demo.launch()