1een commited on
Commit
3bced8d
·
1 Parent(s): 53000f5
Files changed (2) hide show
  1. fixed_app.py +4 -2
  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.2==="
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:"