Update app.py

app.py

@@ -15,7 +15,7 @@ MODEL =  Path("train_log/")
 
 TMP_PREFIX = "/tmp/gradio/"
 TARGET_DIR = f"{TMP_PREFIX}output/"
-EXTENSION = "png"
+EXTENSION = "gif"
 
 def interpolate_image(img_a_path: str, img_b_path: str) -> Tuple[str, str]:
     """
@@ -44,6 +44,7 @@ def interpolate_image(img_a_path: str, img_b_path: str) -> Tuple[str, str]:
                  "--exp", "4"
         ], check=True)
 
+        print("ok")
         # 2) Generate palette
         subprocess.run([
             "ffmpeg", "-y", "-r", "14", "-f", "image2",
@@ -85,15 +86,14 @@ with gr.Blocks(title="RIFE Interpolation") as demo:
 
     with gr.Tab("Upload your images"):
         gr.Markdown("### Upload any two images")
-        user_A = gr.Image(type="filepath", label="Image A")
-        user_B = gr.Image(type="filepath", label="Image B")
-        user_mode = gr.Dropdown(choices=["default"], value="default", label="Mode")
+        input_imageA = gr.Image(type="filepath", label="Image A")
+        input_imageB = gr.Image(type="filepath", label="Image B")
         user_btn = gr.Button("Interpolate")
-        user_out = gr.Image(type="filepath", label="Interpolated GIF")
-        user_path = gr.Textbox(label="Output path", interactive=False)
+        user_out = gr.Image(type="filepath", value=str(TARGET_DIR), label="Interpolated GIF", value=str(DEMO))
+        user_path = gr.Textbox(label="Output path", value=str(TARGET_DIR), interactive=False)
         user_btn.click(
             fn=interpolate_image,
-            inputs=[user_A, user_B],
+            inputs=[input_imageA, input_imageB],
             outputs=[user_out, user_path],
         )