|
|
from fastapi import FastAPI, File, UploadFile |
|
|
from utils.preprocessing import load_audio |
|
|
from utils.inference import get_embeddings, predict_risk |
|
|
|
|
|
app = FastAPI() |
|
|
|
|
|
@app.get("/") |
|
|
def hello(): |
|
|
return {"Hello": "World!"} |
|
|
|
|
|
|
|
|
@app.post("/predict") |
|
|
async def predict(file: UploadFile = File(...)): |
|
|
temp_path = f"/tmp/{file.filename}" |
|
|
with open(temp_path, "wb") as f: |
|
|
f.write(await file.read()) |
|
|
|
|
|
waveform, sr = load_audio(temp_path) |
|
|
embeddings = get_embeddings(waveform, sr) |
|
|
score = predict_risk(embeddings) |
|
|
|
|
|
return {"risk_score": score} |
|
|
|
