|
|
from fastapi import FastAPI, File, UploadFile |
|
|
from fastapi.responses import JSONResponse |
|
|
from transformers import pipeline |
|
|
import uvicorn |
|
|
import tempfile |
|
|
|
|
|
app = FastAPI() |
|
|
|
|
|
|
|
|
classifier = pipeline("audio-classification", |
|
|
model="AmeerHesham/distilhubert-finetuned-baby_cry", |
|
|
device=-1) |
|
|
|
|
|
@app.post("/predict") |
|
|
async def predict_audio(file: UploadFile = File(...)): |
|
|
try: |
|
|
contents = await file.read() |
|
|
with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as tmp: |
|
|
tmp.write(contents) |
|
|
tmp_path = tmp.name |
|
|
|
|
|
results = classifier(tmp_path) |
|
|
return JSONResponse(content={"results": results}) |
|
|
except Exception as e: |
|
|
return JSONResponse(content={"error": str(e)}, status_code=500) |
|
|
|
|
|
if __name__ == "__main__": |
|
|
uvicorn.run("app:app", host="0.0.0.0", port=7860, reload=True) |
|
|
|
