- fixed_app.py +4 -2
- startup.sh +1 -1
fixed_app.py
CHANGED
|
@@ -32,6 +32,7 @@ class AudioRequest(BaseModel):
|
|
| 32 |
beam_size: Optional[int] = 1
|
| 33 |
condition_on_previous_text: Optional[bool] = False
|
| 34 |
fast_mode: Optional[bool] = True # 快速模式
|
|
|
|
| 35 |
|
| 36 |
def load_model(model_name: str):
|
| 37 |
"""确保模型文件存在,返回模型路径"""
|
|
@@ -245,8 +246,8 @@ async def transcribe_audio(request: AudioRequest):
|
|
| 245 |
"-ac", "0", # 音频上下文 = 0 (最快)
|
| 246 |
"-nf", # --no-fallback: 禁用温度回退
|
| 247 |
"-tp", "0.0", # 温度 = 0 (确定性输出)
|
| 248 |
-
"--vad",
|
| 249 |
-
"-vm", "/app/models/ggml-silero-v5.1.2.bin"
|
| 250 |
]
|
| 251 |
else:
|
| 252 |
# 标准模式:平衡速度和精度
|
|
@@ -311,6 +312,7 @@ async def transcribe_audio(request: AudioRequest):
|
|
| 311 |
|
| 312 |
# 读取JSON输出文件
|
| 313 |
result = parse_whisper_output(output_file, stdout, exit_code)
|
|
|
|
| 314 |
|
| 315 |
return result
|
| 316 |
|
|
|
|
| 32 |
beam_size: Optional[int] = 1
|
| 33 |
condition_on_previous_text: Optional[bool] = False
|
| 34 |
fast_mode: Optional[bool] = True # 快速模式
|
| 35 |
+
vad: Optional[bool] = False
|
| 36 |
|
| 37 |
def load_model(model_name: str):
|
| 38 |
"""确保模型文件存在,返回模型路径"""
|
|
|
|
| 246 |
"-ac", "0", # 音频上下文 = 0 (最快)
|
| 247 |
"-nf", # --no-fallback: 禁用温度回退
|
| 248 |
"-tp", "0.0", # 温度 = 0 (确定性输出)
|
| 249 |
+
"--vad" if request.vad else "",
|
| 250 |
+
"-vm", "/app/models/ggml-silero-v5.1.2.bin" if request.vad else ""
|
| 251 |
]
|
| 252 |
else:
|
| 253 |
# 标准模式:平衡速度和精度
|
|
|
|
| 312 |
|
| 313 |
# 读取JSON输出文件
|
| 314 |
result = parse_whisper_output(output_file, stdout, exit_code)
|
| 315 |
+
result["processing_time"] = processing_time
|
| 316 |
|
| 317 |
return result
|
| 318 |
|
startup.sh
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
#!/bin/bash
|
| 2 |
|
| 3 |
# 显示环境信息
|
| 4 |
-
echo "=== Whisper API Startup 0.
|
| 5 |
echo "Python version: $(python3 --version)"
|
| 6 |
echo "Current directory: $(pwd)"
|
| 7 |
# echo "Files in /app:"
|
|
|
|
| 1 |
#!/bin/bash
|
| 2 |
|
| 3 |
# 显示环境信息
|
| 4 |
+
echo "=== Whisper API Startup 0.3==="
|
| 5 |
echo "Python version: $(python3 --version)"
|
| 6 |
echo "Current directory: $(pwd)"
|
| 7 |
# echo "Files in /app:"
|