qfuxa commited on
Commit
9d757f5
·
1 Parent(s): 8d2dff9

remove temp_kit method to get args. uvicorn reload to False for better perfs

Browse files
whisper_fastapi_online_server.py CHANGED
@@ -3,7 +3,7 @@ from fastapi import FastAPI, WebSocket, WebSocketDisconnect
3
  from fastapi.responses import HTMLResponse
4
  from fastapi.middleware.cors import CORSMiddleware
5
 
6
- from whisperlivekit import WhisperLiveKit
7
  from whisperlivekit.audio_processor import AudioProcessor
8
 
9
  import asyncio
@@ -71,12 +71,13 @@ async def websocket_endpoint(websocket: WebSocket):
71
  if __name__ == "__main__":
72
  import uvicorn
73
 
74
- temp_kit = WhisperLiveKit(transcription=False, diarization=False)
75
 
76
  uvicorn.run(
77
  "whisper_fastapi_online_server:app",
78
- host=temp_kit.args.host,
79
- port=temp_kit.args.port,
80
- reload=True,
81
- log_level="info"
 
82
  )
 
3
  from fastapi.responses import HTMLResponse
4
  from fastapi.middleware.cors import CORSMiddleware
5
 
6
+ from whisperlivekit import WhisperLiveKit, parse_args
7
  from whisperlivekit.audio_processor import AudioProcessor
8
 
9
  import asyncio
 
71
  if __name__ == "__main__":
72
  import uvicorn
73
 
74
+ args = parse_args()
75
 
76
  uvicorn.run(
77
  "whisper_fastapi_online_server:app",
78
+ host=args.host,
79
+ port=args.port,
80
+ reload=False,
81
+ log_level="info",
82
+ lifespan="on",
83
  )
whisperlivekit/basic_server.py CHANGED
@@ -3,7 +3,7 @@ from fastapi import FastAPI, WebSocket, WebSocketDisconnect
3
  from fastapi.responses import HTMLResponse
4
  from fastapi.middleware.cors import CORSMiddleware
5
 
6
- from whisperlivekit import WhisperLiveKit
7
  from whisperlivekit.audio_processor import AudioProcessor
8
 
9
  import asyncio
@@ -72,14 +72,15 @@ def main():
72
  """Entry point for the CLI command."""
73
  import uvicorn
74
 
75
- temp_kit = WhisperLiveKit(transcription=False, diarization=False)
76
 
77
  uvicorn.run(
78
  "whisperlivekit.basic_server:app",
79
- host=temp_kit.args.host,
80
- port=temp_kit.args.port,
81
- reload=True,
82
- log_level="info"
 
83
  )
84
 
85
  if __name__ == "__main__":
 
3
  from fastapi.responses import HTMLResponse
4
  from fastapi.middleware.cors import CORSMiddleware
5
 
6
+ from whisperlivekit import WhisperLiveKit, parse_args
7
  from whisperlivekit.audio_processor import AudioProcessor
8
 
9
  import asyncio
 
72
  """Entry point for the CLI command."""
73
  import uvicorn
74
 
75
+ args = parse_args()
76
 
77
  uvicorn.run(
78
  "whisperlivekit.basic_server:app",
79
+ host=args.host,
80
+ port=args.port,
81
+ reload=False,
82
+ log_level="info",
83
+ lifespan="on",
84
  )
85
 
86
  if __name__ == "__main__":
whisperlivekit/whisper_streaming_custom/whisper_online.py CHANGED
@@ -179,7 +179,7 @@ def warmup_asr(asr, warmup_file=None, timeout=5):
179
  logger.warning(f"Warmup file {warmup_file} invalid or missing.")
180
  return False
181
 
182
- print(f"Warmping up Whisper with {warmup_file}")
183
  try:
184
  import librosa
185
  audio, sr = librosa.load(warmup_file, sr=16000)
 
179
  logger.warning(f"Warmup file {warmup_file} invalid or missing.")
180
  return False
181
 
182
+ print(f"Warming up Whisper with {warmup_file}")
183
  try:
184
  import librosa
185
  audio, sr = librosa.load(warmup_file, sr=16000)