Upload folder using huggingface_hub

app.py

@@ -24,20 +24,9 @@ def gpu_decorator(func):
         return GPU(func)
     return func
 
-# 条件安装 flash-attn（仅在 GPU 可用时尝试安装）
-if is_spaces and torch.cuda.is_available():
-    try:
-        import subprocess
-        print("🔧 Installing flash-attn for GPU acceleration...")
-        subprocess.run(
-            'pip install flash-attn --no-build-isolation', 
-            env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, 
-            shell=True,
-            check=False
-        )
-        print("✅ flash-attn installation completed")
-    except Exception as e:
-        print(f"⚠️ flash-attn installation failed (will use eager attention): {e}")
+# 条件安装 flash-attn（延迟到模型加载时，避免启动时 CUDA 检查）
+# 注意：在 ZeroGPU 环境中，启动时 CUDA 可能还不可用
+# flash-attn 将在模型加载时根据实际 CUDA 可用性决定是否使用
 
 sys_prompt = """First output the the types of degradations in image briefly in <TYPE> <TYPE_END> tags, 
         and then output what effects do these degradation have on the image in <INFLUENCE> <INFLUENCE_END> tags, 
@@ -65,7 +54,6 @@ class ModelHandler:
         self.processor = None
         self._load_model()
 
-    @gpu_decorator  # 标记此函数需要 GPU
     def _load_model(self):
         try:
             print(f"⏳ Loading model weights, this may take a few minutes...")
@@ -322,19 +310,7 @@ def create_chat_ui():
 
     return demo
 
-# 在启动时声明需要 GPU（用于 ZeroGPU 检测）
-@gpu_decorator
-def initialize_gpu():
-    """初始化 GPU，用于 Spaces 检测"""
-    if is_spaces:
-        print("✅ GPU declared for Hugging Face Spaces")
-    return True
-
 if __name__ == "__main__":
-    # 在启动时调用以声明 GPU
-    if is_spaces:
-        initialize_gpu()
-    
     demo = create_chat_ui()
     
     custom_css = """