Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import torch
 import cv2
@@ -22,6 +23,7 @@ import subprocess
 import sys
 
 
+@spaces.GPU(duration=120)
 def download_liveportrait():
     """
     Clone the LivePortrait repository and prepare its dependencies.
@@ -57,6 +59,7 @@ def download_liveportrait():
         raise
 download_liveportrait()
 
+@spaces.GPU(duration=120)
 def download_huggingface_resources():
     """
     Download additional necessary resources from Hugging Face using the CLI.
@@ -85,6 +88,8 @@ def download_huggingface_resources():
 
 download_huggingface_resources()
 
+
+@spaces.GPU(duration=120)
 def get_project_root():
     """Get the root directory of the current project."""
     return os.path.abspath(os.path.dirname(__file__))
@@ -121,10 +126,12 @@ pipe_inpaint_controlnet.to(device)
 pipe_inpaint_controlnet.enable_attention_slicing()
 
 
+@spaces.GPU(duration=120)
 def resize_to_multiple_of_64(width, height):
     return (width // 64) * 64, (height // 64) * 64
 
 
+@spaces.GPU(duration=120)
 def expand_mask(mask, kernel_size):
     mask_array = np.array(mask)
     structuring_element = np.ones((kernel_size, kernel_size), dtype=np.uint8)
@@ -134,6 +141,7 @@ def expand_mask(mask, kernel_size):
     return Image.fromarray(expanded_mask_array)
 
 
+@spaces.GPU(duration=120)
 def crop_face_to_square(image_rgb, padding_ratio=0.2):
     """
     Detects the face in the input image and crops an enlarged square region around it.
@@ -163,6 +171,7 @@ def crop_face_to_square(image_rgb, padding_ratio=0.2):
     return resized_image
 
 
+@spaces.GPU(duration=120)
 def spirit_animal_baseline(image_path, num_images = 4):
 
     image = cv2.imread(image_path)
@@ -244,6 +253,7 @@ def spirit_animal_baseline(image_path, num_images = 4):
     return prompt, generated_images
 
 
+@spaces.GPU(duration=120)
 def spirit_animal_with_background(image_path, num_images = 4):
 
     image = cv2.imread(image_path)
@@ -333,6 +343,7 @@ def spirit_animal_with_background(image_path, num_images = 4):
     return prompt, generated_images
 
 
+@spaces.GPU(duration=120)
 def generate_multiple_animals(image_path, keep_background=True, num_images = 4):
 
     image = cv2.imread(image_path)
@@ -454,6 +465,7 @@ def generate_multiple_animals(image_path, keep_background=True, num_images = 4):
     return formatted_prompts, generated_images
 
 
+@spaces.GPU(duration=120)
 def wait_for_file(file_path, timeout=500):
     """
     Wait for a file to be created, with a specified timeout.
@@ -471,6 +483,7 @@ def wait_for_file(file_path, timeout=500):
     return True
 
 
+@spaces.GPU(duration=120)
 def generate_spirit_animal_video(driving_video_path):
     os.chdir(".")
     try:
@@ -529,6 +542,7 @@ def generate_spirit_animal_video(driving_video_path):
         return None
 
 
+@spaces.GPU(duration=120)
 def generate_spirit_animal(image, animal_type, background):
     if animal_type == "Single Animal":
         if background == "Preserve Background":
@@ -543,6 +557,7 @@ def generate_spirit_animal(image, animal_type, background):
     return prompt, generated_images
 
 
+@spaces.GPU(duration=120)
 def compress_video(input_path, output_path, target_size_mb):
     target_size_bytes = target_size_mb * 1024 * 1024
     temp_output = "./temp_compressed.mp4"
@@ -572,6 +587,7 @@ def compress_video(input_path, output_path, target_size_mb):
         shutil.move(temp_output, output_path)
 
 
+@spaces.GPU(duration=120)
 def process_video(video_file):
     
     # # 初始化 LivePortrait