42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
|
import os
|
||
|
|
import tempfile
|
||
|
|
|
||
|
|
from faster_whisper import WhisperModel
|
||
|
|
from fastapi import FastAPI, File, Form, UploadFile
|
||
|
|
from fastapi.responses import JSONResponse
|
||
|
|
|
||
|
|
app = FastAPI()
|
||
|
|
model: WhisperModel | None = None
|
||
|
|
|
||
|
|
|
||
|
|
@app.on_event("startup")
|
||
|
|
def load_model():
|
||
|
|
global model
|
||
|
|
model_size = os.environ.get("WHISPER_MODEL", "tiny")
|
||
|
|
model = WhisperModel(model_size, device="cpu", compute_type="int8")
|
||
|
|
|
||
|
|
|
||
|
|
@app.post("/transcribe")
|
||
|
|
async def transcribe(
|
||
|
|
file: UploadFile = File(...),
|
||
|
|
language: str = Form(None),
|
||
|
|
):
|
||
|
|
if model is None:
|
||
|
|
return JSONResponse(status_code=503, content={"error": "Model not loaded"})
|
||
|
|
|
||
|
|
with tempfile.NamedTemporaryFile(suffix=".wav", delete=True) as tmp:
|
||
|
|
tmp.write(await file.read())
|
||
|
|
tmp.flush()
|
||
|
|
kwargs = {}
|
||
|
|
if language:
|
||
|
|
kwargs["language"] = language
|
||
|
|
segments, info = model.transcribe(tmp.name, **kwargs)
|
||
|
|
text = " ".join(s.text for s in segments).strip()
|
||
|
|
|
||
|
|
return {"text": text, "language": info.language}
|
||
|
|
|
||
|
|
|
||
|
|
@app.get("/health")
|
||
|
|
def health():
|
||
|
|
return {"status": "ok"}
|