Update app.py

app.py

@@ -1,4 +1,3 @@
-import spaces
 import gradio as gr
 import torch
 import cv2
@@ -22,24 +21,7 @@ from huggingface_hub import snapshot_download
 import subprocess
 import sys
 
-@spaces.GPU(duration=120)
-def install_deformable_detr():
-    # 克隆 Deformable-DETR 仓库
-    if not os.path.exists('Deformable-DETR'):
-        subprocess.run(['git', 'clone', 'https://github.com/fundamentalvision/Deformable-DETR.git'])
-    
-    # 安装依赖
-    subprocess.run(['pip', 'install', '-r', 'Deformable-DETR/requirements.txt'])
-    
-    # 编译 CUDA 操作符
-    os.chdir('Deformable-DETR/models/ops')
-    subprocess.run(['sh', './make.sh'])
-    os.chdir('../../../..')
-
-# 在应用启动时调用安装函数
-install_deformable_detr()
 
-@spaces.GPU(duration=120)
 def download_liveportrait():
     """
     Clone the LivePortrait repository and prepare its dependencies.
@@ -74,7 +56,9 @@ def download_liveportrait():
         print("Failed to initialize LivePortrait:", e)
         raise
 
-@spaces.GPU(duration=120)
+download_liveportrait()
+
+
 def download_huggingface_resources():
     """
     Download additional necessary resources from Hugging Face using the CLI.
@@ -101,7 +85,9 @@ def download_huggingface_resources():
         print("General error in downloading resources:", e)
         raise
 
-@spaces.GPU(duration=120)
+download_huggingface_resources():
+
+
 def get_project_root():
     """Get the root directory of the current project."""
     return os.path.abspath(os.path.dirname(__file__))
@@ -137,11 +123,11 @@ pipe_controlnet.enable_attention_slicing()
 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)
@@ -150,7 +136,7 @@ def expand_mask(mask, kernel_size):
     ).astype(np.uint8) * 255
     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.
@@ -179,7 +165,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)
@@ -260,7 +246,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)
@@ -350,7 +336,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)
@@ -471,7 +456,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.
@@ -488,7 +473,7 @@ def wait_for_file(file_path, timeout=500):
         time.sleep(0.5)  # Check every 0.5 seconds
     return True
 
-@spaces.GPU(duration=120)
+
 def generate_spirit_animal_video(driving_video_path):
     os.chdir(".")
     try:
@@ -594,19 +579,19 @@ def compress_video(input_path, output_path, target_size_mb):
 @spaces.GPU(duration=120)
 def process_video(video_file):
     
-    # 初始化 LivePortrait
-    try:
-        download_liveportrait()
-    except Exception as e:
-        print("Failed to initialize LivePortrait:", e)
-        return gr.update(value=None, visible=False)
+    # # 初始化 LivePortrait
+    # try:
+    #     download_liveportrait()
+    # except Exception as e:
+    #     print("Failed to initialize LivePortrait:", e)
+    #     return gr.update(value=None, visible=False)
     
-        # 下载 Hugging Face 资源
-    try:
-        download_huggingface_resources()
-    except Exception as e:
-        print("Failed to download Hugging Face resources:", e)
-        return gr.update(value=None, visible=False)
+    #     # 下载 Hugging Face 资源
+    # try:
+    #     download_huggingface_resources()
+    # except Exception as e:
+    #     print("Failed to download Hugging Face resources:", e)
+    #     return gr.update(value=None, visible=False)
     
     compressed_path = "./uploaded_video_compressed.mp4"
     compress_video(video_file, compressed_path, target_size_mb=1)