Update app.py

app.py

@@ -6,6 +6,7 @@ The entrance of the gradio
 
 import tyro
 import gradio as gr
+import spaces
 import os.path as osp
 from src.utils.helper import load_description
 from src.gradio_pipeline import GradioPipeline
@@ -32,7 +33,8 @@ gradio_pipeline = GradioPipeline(
     args=args
 )
 
-def gpu_wrapped_execute_s_video(*args, **kwargs):
+@spaces.GPU()
+def gpu_wrapped_execute_s_video(*args, **kwargs, progress=gr.Progress()):
     return gradio_pipeline.execute_s_video(*args, **kwargs)
 
 # assets
@@ -44,22 +46,17 @@ example_video_dir = "assets/examples/driving"
 
 # Define components first
 
-retargeting_input_image = gr.Image(type="filepath")
-output_image = gr.Image(type="numpy")
-output_image_paste_back = gr.Image(type="numpy")
-output_video = gr.Video()
-output_video_concat = gr.Video()
-
 output_video1 = gr.Video()
 output_video_concat1 = gr.Video()
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.HTML(load_description(title_md))
-    gr.Markdown(load_description("assets/gradio_description_upload.md"))
     
     with gr.Row():
         # Examples
-        gr.Markdown("## For video to video")
+        gr.Markdown("""
+        ## 🤗 This is the gradio demo for LivePortrait for video.
+        Please upload or use a webcam to get a Source Portrait Video (any aspect ratio) and upload a Driving Video (1:1 aspect ratio, or any aspect ratio with do crop (driving video) checked).
+        """)
     # for video portrait
     with gr.Row():
         with gr.Accordion(open=True, label="Video Portrait"):