modify: change for space.GPU

app.py

@@ -6,16 +6,41 @@ import cv2
 import zipfile
 from utils import process_video, get_npy_files, get_frame_count, process_image
 from infer_script import run_inference
-import huggingface_hub.spaces as spaces
 import time
 import datetime
 import shutil
+import torch
 
 import imageio
 from media_pipe.draw_util import FaceMeshVisualizer
 
 from download_models import download 
 
+def check_gpu():
+    if torch.cuda.is_available():
+        device_name = torch.cuda.get_device_name(0)
+        print(f"使用 GPU: {device_name}")
+        return "cuda"
+    else:
+        print("GPU 不可用，使用 CPU")
+        return "cpu"
+
+# 设置全局设备
+DEVICE = check_gpu()
+
+try:
+    from huggingface_hub import spaces
+    HAS_SPACES = True
+except ImportError:
+    print("Warning: huggingface_hub.spaces module not available, GPU acceleration features may be limited")
+    HAS_SPACES = False
+    
+    # 创建一个替代装饰器
+    class SpacesMock:
+        def GPU(self, func):
+            return func
+    spaces = SpacesMock()
+
 # Download models and check for exists
 download()
 
@@ -170,7 +195,6 @@ def preview_crop(image_path, npy_file, video_path, expand_x, expand_y, offset_x,
     else:
         return None,None, "Failed to generate crop preview"
 
-@spaces.GPU
 def generate_video(input_img, should_crop_face, expand_x, expand_y, offset_x, offset_y, input_video_type, input_video, input_npy_select, input_npy, input_video_frames,
                    settings_steps, settings_cfg_scale, settings_seed, resolution_w, resolution_h,
                    model_step, custom_output_path, use_custom_fps, output_fps, callback_steps, context_frames, context_stride, context_overlap, context_batch_size, anomaly_action,intropolate_factor):
@@ -234,7 +258,8 @@ def generate_video(input_img, should_crop_face, expand_x, expand_y, offset_x, of
         context_overlap=context_overlap,
         context_batch_size=context_batch_size,
         anomaly_action=anomaly_action,
-        interpolation_factor=intropolate_factor
+        interpolation_factor=intropolate_factor,
+        device=DEVICE
     )
 
 

requirements.txt

@@ -24,7 +24,6 @@ protobuf==4.25.3
 xformers==0.0.23
 gradio
 scikit-image
-huggingface_hub==0.20.0
 # extra dependencies
 torch==2.1.1
 torchaudio==2.1.1