updated app

app.py

@@ -136,19 +136,29 @@ with gr.Blocks() as demo:
 
     with gr.Tab("Image Inference"):
         with gr.Row():
-            source_img = gr.Image(type="pil")
-            driver_img =  gr.Image(type="pil")
+            source_img = gr.Image(type="pil", label="source image", show_label=True)
+            driver_img =  gr.Image(type="pil", label="driver image", show_label=True)
         image_output = gr.Image()
         image_button = gr.Button("Predict")
     with gr.Tab("Video Inference"):
-        video_inputs = [gr.Video(), gr.Image()]
-        pass
+        with gr.Row():
+            source_video = gr.Video(label="source video", )
+            driver_image_for_vid = gr.Image(type="pil", label="driver image", show_label=True)
+        video_output = gr.Image()
+        video_button = gr.Button("Predict")
 
     gr.Examples(
-        examples=[["./examples/lincoln.jpg", "./examples/taras2.jpg"]],
-        inputs=[source_img, driver_img]
+        examples=[
+            ["./examples/lincoln.jpg", "./examples/taras2.jpg"],
+            ["./examples/lincoln.jpg", "./examples/taras1.jpg"]
+        ],
+        inputs=[source_img, driver_img],
+        outputs=[image_output],
+        fn=image_inference,
+        cache_examples=True
     )
 
     image_button.click(image_inference, inputs=[source_img, driver_img], outputs=image_output)
+    video_button.click(None, inputs=[source_video, driver_image_for_vid], outputs=video_output)
 
 demo.launch()