update: app.py, gpen for video

app.py

@@ -440,18 +440,20 @@ def swap_video_gr(img1, target_path, use_gpen=False, frames=9999999):
     return video_save_path
 
 
-css = """
-#warning {background-color: #FFCCCB} 
-.feedback textarea {font-size: 24px !important}
-.run_button {background-color: orange} 
-"""
-
-
 if __name__ == "__main__":
     use_gpu = torch.cuda.is_available()
 
     with gr.Blocks() as demo:
-        gr.Markdown("SuperSwap")
+        gr.Markdown("<div align='center'> <h2> ReliableSwap: Boosting General Face Swapping Via Reliable Supervision </span> </h2> \
+                     <a style='font-size:18px;color: #efefef' href='https://arxiv.org/abs/xxxx.xxxx'>Arxiv</a> &nbsp;&nbsp;&nbsp;&nbsp;&nbsp; \
+                     <a style='font-size:18px;color: #efefef' href='https://reliable-swap.github.io/'>Homepage</a>  &nbsp;&nbsp;&nbsp;&nbsp;&nbsp; \
+                     <a style='font-size:18px;color: #efefef' href='https://github.com/ygtxr1997/ReliableSwap'> Github </div>")
+
+        gr.Markdown("""
+                <b>Alternatively, try our GitHub <a href=https://github.com/Winfredy/SadTalker> code </a> on your own GPU. </b> 
+                <a style='display:inline-block' href="https://github.com/ygtxr1997/ReliableSwap">
+                    <img src="https://img.shields.io/github/stars/ygtxr1997/ReliableSwap?style=social"/></a> \
+                """)
 
         with gr.Tab("Image"):
             with gr.Row():
@@ -459,10 +461,10 @@ if __name__ == "__main__":
                     image1_input = gr.Image(label='source')
                     image2_input = gr.Image(label='target')
                     image_use_post = gr.Checkbox(label="Post-Process")
-                    image_use_gpen = gr.Checkbox(label="Super Resolution")
+                    image_use_gpen = gr.Checkbox(label="Super Resolution (based on GPEN)")
                 with gr.Column(scale=2):
                     image_output = gr.Image()
-                    image_button = gr.Button("Run: Face Swapping", elem_classes="run_button")
+                    image_button = gr.Button("Run: Face Swapping", variant="primary")
         with gr.Tab("Video"):
             with gr.Row():
                 with gr.Column(scale=3):
@@ -470,8 +472,9 @@ if __name__ == "__main__":
                     video_input = gr.Video(label='target')
                 with gr.Column(scale=2):
                     video_output = gr.Video()
-                    video_use_gpen = gr.Checkbox(label="Super Resolution")
-                    video_button = gr.Button("Run: Face Swapping", elem_classes="run_button")
+                    video_use_gpen = gr.Checkbox(label="Super Resolution (based on GPEN). "
+                                                       "Warning: very time-cost on CPU!")
+                    video_button = gr.Button("Run: Face Swapping", variant="primary")
         image_button.click(
             swap_image_gr,
             inputs=[image1_input, image2_input, image_use_post, image_use_gpen],