Update app.py

app.py

@@ -4,6 +4,7 @@ import logging
 import re
 import gradio as gr
 from spaces import zero  # 关键：引入 zero 装饰器
+import spaces
 
 # 不要在这里 import torch 或加载模型
 # from transformers import TextIteratorStreamer, AutoTokenizer  # 不再需要
@@ -29,7 +30,7 @@ def _str_to_dtype(dtype_str):
         return dtype_str
     return "float32"
 
-@zero.gpu  # 在子进程（拥有 GPU）中执行：包含模型加载与推理
+@spaces.GPU  # 在子进程（拥有 GPU）中执行：包含模型加载与推理
 def gpu_predict(model_path, device_map, torch_dtype,
                 prompt_cot, sys_prompt, temperature, max_new_tokens, device):
     # 注意：所有 CUDA 相关 import 放在子进程函数内部