Update app_.py

app_.py

@@ -152,6 +152,11 @@ def gpu_wrapped_execute_video(*args, **kwargs):
 def gpu_wrapped_execute_image(*args, **kwargs):
     return gradio_pipeline.execute_image(*args, **kwargs)
 
+@spaces.GPU()
+def gpu_wrapped_execute_retargeting(*args, **kwargs):
+    return gradio_pipeline.prepare_retargeting(*args, **kwargs)
+    
+
 def is_square_video(video_path):
     video = cv2.VideoCapture(video_path)
 
@@ -331,7 +336,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     )
 
     process_button_closelip.click(
-        fn=gradio_pipeline.prepare_retargeting,
+        fn=gpu_wrapped_execute_retargeting,
         inputs=image_input,
         outputs=image_input,
         show_progress=True