updated app

app.py

@@ -198,14 +198,12 @@ def video_inference(
     
     return "avatar.gif"
 
-
-teaser = "<p style='text-align: center'> <img src='file/media/tease.gif' alt='Teaser'> </p>"
-description = "<p style='text-align: center'> Create a personal avatar from just a single image using ROME. <br> <a href='https://arxiv.org/abs/2206.08343' target='_blank'>Paper</a> | <a href='https://samsunglabs.github.io/rome' target='_blank'>Project Page</a> | <a href='https://github.com/SamsungLabs/rome' target='_blank'>Github</a> </p>"
-quote = "<p style='text-align: center'> <blockquote> [The] system creates realistic mesh-based avatars from a single <strong>source</strong> photo. These avatars are rigged, i.e., they can be driven by the animation parameters from a different <strong>driving</strong> frame. </blockquote> </p>"
+description = """<p style='text-align: center'> Create a personal avatar from just a single image using ROME. <br> <a href='https://arxiv.org/abs/2206.08343' target='_blank'>Paper</a> | <a href='https://samsunglabs.github.io/rome' target='_blank'>Project Page</a> | <a href='https://github.com/SamsungLabs/rome' target='_blank'>Github</a> </p>"""
+quote = """<p style='text-align: center'> <blockquote> [The] system creates realistic mesh-based avatars from a single <strong>source</strong> photo. These avatars are rigged, i.e., they can be driven by the animation parameters from a different <strong>driving</strong> frame. </blockquote> </p>"""
 
 with gr.Blocks() as demo:
     gr.Markdown("# **<p align='center'>ROME: Realistic one-shot mesh-based head avatars</p>**")
-    gr.HTML(teaser)
+    gr.HTML(value="<img src='file/media/tease.gif' alt='Teaser' style='display: block; margin: auto;'>")
     gr.Markdown(description)
     gr.HTML(quote)
 
@@ -217,8 +215,8 @@ with gr.Blocks() as demo:
         image_button = gr.Button("Predict")
     with gr.Tab("Video Inference"):
         with gr.Row():
-            source_img2 = gr.Image(type="pil", label="source image", show_label=True)
-            driver_vid = gr.Video(label="driver video")
+            source_img2 = gr.Image(type="pil", label="Source image", show_label=True)
+            driver_vid = gr.Video(label="Driver video")
         video_output = gr.Image(label="Rendered GIF avatar")
         video_button = gr.Button("Predict")