Update app.py

app.py

@@ -8,11 +8,10 @@ from fastapi import FastAPI, UploadFile, File, Form, HTTPException
 from fastapi.responses import JSONResponse, PlainTextResponse
 from transformers import pipeline
 
-# 1. 声明加载的模型
+# 1. 模型加载
 MODEL_NAME = "openai/whisper-small"
-
-# 2. 初始化 Pipeline
 device = "cuda" if torch.cuda.is_available() else "cpu"
+
 pipe = pipeline(
     "automatic-speech-recognition",
     model=MODEL_NAME,
@@ -20,7 +19,7 @@ pipe = pipeline(
     device=device
 )
 
-# 3. 【关键点】在最外层定义带 @spaces.GPU 的核心推理函数
+# 2. 核心 GPU 推理函数（ZeroGPU 严格要求必须在文件顶层声明）
 @spaces.GPU
 def run_whisper_inference(audio_path: str, target_language: str = None, is_translate: bool = False):
     generate_kwargs = {}
@@ -34,8 +33,7 @@ def run_whisper_inference(audio_path: str, target_language: str = None, is_trans
     result = pipe(audio_path, generate_kwargs=generate_kwargs)
     return result["text"]
 
-
-# --- Gradio 界面 ---
+# 3. Gradio UI 封装
 def gradio_predict(audio_path):
     if audio_path is None:
         return "请先上传音频或录音！"
@@ -52,8 +50,7 @@ demo = gr.Interface(
     description="【完美兼容 OpenAI 规范】"
 )
 
-
-# --- FastAPI 接口配置 ---
+# 4. FastAPI 应用声明
 app = FastAPI()
 
 async def process_openai_audio_request(file: UploadFile, response_format: str, language: str, is_translate: bool):
@@ -63,7 +60,6 @@ async def process_openai_audio_request(file: UploadFile, response_format: str, l
         temp_path = temp_file.name
 
     try:
-        # 调用最外层带 @spaces.GPU 装饰的函数
         text = run_whisper_inference(temp_path, target_language=language, is_translate=is_translate)
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Inference failed: {str(e)}")
@@ -76,7 +72,6 @@ async def process_openai_audio_request(file: UploadFile, response_format: str, l
 
     return JSONResponse(content={"text": text})
 
-
 @app.post("/v1/audio/transcriptions")
 async def transcribe_api(
     file: UploadFile = File(...),
@@ -93,7 +88,6 @@ async def transcribe_api(
         is_translate=False
     )
 
-
 @app.post("/v1/audio/translations")
 async def translate_api(
     file: UploadFile = File(...),
@@ -109,5 +103,5 @@ async def translate_api(
         is_translate=True
     )
 
-# 挂载 Gradio 页面到 FastAPI 根路径
+# 5. 挂载 Gradio 到 FastAPI
 app = gr.mount_gradio_app(app, demo, path="/")