updated app

app.py

@@ -132,21 +132,28 @@ def video_inference():
     pass
 
 with gr.Blocks() as demo:
+    gr.Markdown("# **<p align='center'>ROME: Realistic one-shot mesh-based head avatars</p>**")
+
     with gr.Tab("Image Inference"):
         with gr.Row():
-            image_inputs = [gr.Image(type="pil"), gr.Image(type="pil")]
-            image_output = gr.Image()
+            source_img = gr.Image(type="pil")
+            driver_img =  gr.Image(type="pil")
+        image_output = gr.Image()
         image_button = gr.Button("Predict")
     with gr.Tab("Video Inference"):
         video_inputs = [gr.Video(), gr.Image()]
         pass
-
-    title = "ROME: Realistic one-shot mesh-based head avatars"
-    examples = gr.Examples(
-        examples="./examples",
-        inputs=image_inputs
+    
+    gr.Examples(
+        examples=[
+            ["./examples/lincoln.jpg"],
+            ["./examples/lincoln.jpg"]
+        ],
+        inputs=[source_img, driver_img],
+        fn=image_inference,
+        cache_examples=True
     )
 
-    image_button.click(image_inference, inputs=image_inputs, outputs=image_output)
+    image_button.click(image_inference, inputs=[source_img, driver_img], outputs=image_output)
 
 demo.launch()

examples/log.csv

@@ -1 +0,0 @@
-"examples/lincoln.jpg", "examples/taras2.jpg"