vlad.os/simple-asr-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

standard model is turbo now

Browse Source
This commit is contained in:
vladislav
2025-07-11 20:30:58 +03:00
parent d6aea35a14
commit a99053afdb
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app.py
View File

@@ -26,7 +26,7 @@ model = None
def load_model(): def load_model():
global model global model
logger.info("Loading whisper model...") logger.info("Loading whisper model...")
model = whisper.load_model("medium", device="cuda", in_memory=True) model = whisper.load_model("turbo", device="cuda", in_memory=True)
logger.info("Whisper model loaded.") logger.info("Whisper model loaded.")
@@ -119,7 +119,7 @@ def get_audio_duration(file_path: str) -> float:
async def transcribe_audio( async def transcribe_audio(
file: UploadFile = File(...), file: UploadFile = File(...),
token: str = Depends(api_key_header), token: str = Depends(api_key_header),
model_name: str = "medium", model_name: str = "turbo",
verbose: Optional[bool] = None, verbose: Optional[bool] = None,
temperature: Union[float, Tuple[float, ...]] = (0.0, 0.2, 0.4, 0.6, 0.8, 1.0), temperature: Union[float, Tuple[float, ...]] = (0.0, 0.2, 0.4, 0.6, 0.8, 1.0),
compression_ratio_threshold: Optional[float] = 2.4, compression_ratio_threshold: Optional[float] = 2.4,