Update app.py

app.py

@@ -1,4 +1,3 @@
-import spaces
 import gradio as gr
 import torch
 import cv2
@@ -23,7 +22,6 @@ import subprocess
 import sys
 
 
-@spaces.GPU(duration=120)
 def download_liveportrait():
     """
     Clone the LivePortrait repository and prepare its dependencies.
@@ -59,7 +57,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.
@@ -89,7 +87,6 @@ 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__))
@@ -126,12 +123,10 @@ 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)
@@ -141,7 +136,6 @@ 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.
@@ -171,7 +165,6 @@ 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)
@@ -253,7 +246,6 @@ 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)
@@ -343,7 +335,6 @@ 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)
@@ -465,7 +456,6 @@ 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.
@@ -483,7 +473,6 @@ 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:
@@ -542,7 +531,6 @@ 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":
@@ -557,7 +545,6 @@ 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"
@@ -587,7 +574,6 @@ 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
@@ -613,7 +599,7 @@ def process_video(video_file):
     generate_spirit_animal_video(compressed_path)
 
     # Wait until the output video is generated
-    timeout = 60000  # Timeout in seconds
+    timeout = 1000  # Timeout in seconds
     if not wait_for_file(output_video_path, timeout=timeout):
         print("Timeout occurred while waiting for video generation.")
         return gr.update(value=None, visible=False)  # Hide output if failed