Update app.py

app.py

@@ -89,11 +89,14 @@ def create_video(frames, fps, type):
     return type + "_result.mp4"
 
     
-def infer(video_in,interpolation,fps_output):
+def infer(url_in,video_in,interpolation,fps_output):
     
     
     # 1. break video into frames and get FPS
-    break_vid = get_frames(video_in, "vid_input_frame", "origin")
+    if url_in.value == "":
+        break_vid = get_frames(video_in, "vid_input_frame", "origin")
+    else:
+        break_vid = get_frames(url_in, "vid_input_frame", "origin")
     frames_list= break_vid[0]
     fps = break_vid[1]
     print(f"ORIGIN FPS: {fps}")
@@ -159,8 +162,8 @@ with gr.Blocks() as demo:
         gr.HTML(title)
         with gr.Row():
             with gr.Column():
-                #video_input = gr.Video(source="upload", type="filepath")
-                video_input = gr.Textbox(value="./examples/no_framerate.mp4")
+                video_input = gr.Video(source="upload", type="filepath")
+                url_input = gr.Textbox(value="./examples/no_framerate.mp4")
                 with gr.Row():
                     interpolation = gr.Slider(minimum=1,maximum=4,step=1, value=1, label="Interpolation Steps")
                     fps_output = gr.Radio([8, 12, 24], label="FPS output", value=8)
@@ -173,11 +176,11 @@ with gr.Blocks() as demo:
     gr.Examples(
         examples=[["./examples/no_framerate.mp4", 1, 8]],
         fn=infer,
-        inputs=[video_input,interpolation,fps_output],
+        inputs=[url_input,video_input,interpolation,fps_output],
         outputs=[video_output,file_output],
         cache_examples=True
     )
     
-    submit_btn.click(fn=infer, inputs=[video_input,interpolation,fps_output], outputs=[video_output, file_output])
+    submit_btn.click(fn=infer, inputs=[url_input,video_input,interpolation,fps_output], outputs=[video_output, file_output])
 
 demo.launch()