Add GPU support via spaces module

app.py

@@ -42,6 +42,21 @@ except Exception as e:
     print(f"Error importing gradio: {e}")
     GRADIO_AVAILABLE = False
 
+# HuggingFace Spaces GPU 支持
+try:
+    import spaces
+    GPU_AVAILABLE = True
+    print("GPU support enabled via spaces module")
+except ImportError:
+    GPU_AVAILABLE = False
+    print("spaces module not available (CPU mode)")
+    # 创建假的装饰器
+    class FakeSpaces:
+        @staticmethod
+        def GPU(func):
+            return func
+    spaces = FakeSpaces()
+
 
 # =============================================================================
 # 配置
@@ -196,8 +211,9 @@ def update_sample_gallery(task: str, combo: str):
     return images[0][0], f"Sample: {sample_info['subject_id']} | Label: {sample_info.get('diag_group', 'N/A')}"
 
 
+@spaces.GPU
 def run_inference(task: str, combo: str):
-    """执行推理"""
+    """执行推理 (GPU accelerated)"""
     engine = get_inference_engine()
     
     if not engine:

requirements.txt

@@ -1,16 +1,16 @@
-# BrainAnytime Hugging Face Space Requirements
+# Brainanytime Hugging Face Space Requirements
 #
-# IMPORTANT: Gradio is pre-installed by Hugging Face (version controlled via README.md sdk_version)
-# Do NOT include gradio here to avoid version conflicts
-#
-# NOTE: huggingface-hub is also pre-installed, we use the Space-provided version
-# to avoid ImportError with HfFolder
+# IMPORTANT: Gradio and huggingface-hub are pre-installed by Hugging Face
+# Do NOT include them here to avoid version conflicts
 
 # Core dependencies
 torch>=2.0.0
 torchvision>=0.15.0
 numpy>=1.24.0
 
+# HuggingFace Spaces GPU support
+spaces>=0.30.0
+
 # Medical imaging
 nibabel>=4.0.0
 torchio>=0.18.95