Update app.py

app.py

@@ -38,7 +38,17 @@ if HF_TOKEN:
 # 在 GPU 子进程 import 阶段就把大模型读进显存
 # 只在 GPU 进程执行，CPU 主进程跳过
 # ─────────────────────────────────────────────
-if spaces.GPU.is_gpu():
+
+# 检测是否在 GPU 环境中运行
+def is_gpu_available():
+    """检测当前环境是否支持 GPU"""
+    try:
+        import torch
+        return torch.cuda.is_available()
+    except ImportError:
+        return False
+
+if is_gpu_available():
     from diffusers import FluxPipeline, FluxTransformer2DModel
     from transformers import AutoModelForCausalLM, AutoTokenizer
     
@@ -79,7 +89,7 @@ if spaces.GPU.is_gpu():
 # ─────────────────────────────────────────────
 def enhance_prompt_with_qwen(original_prompt):
     """使用预加载的 Qwen 模型增强提示词"""
-    if not spaces.GPU.is_gpu():
+    if not is_gpu_available():
         return original_prompt
         
     prompt_template = """You are an expert poster prompt designer. Your task is to rewrite a user's short poster prompt into a detailed and vivid long-format prompt. Follow these steps carefully: