Spaces:
Sleeping
Sleeping
File size: 511 Bytes
e5aa1e6 aabc2fd 4cee7eb e5aa1e6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
from fastapi import FastAPI, HTTPException
from app.model.model import predict_language, __version__
from app.schemas import Prediction, TextIn
app = FastAPI()
@app.get("/")
def home():
return {"message": "Ok", "model_version": __version__}
@app.post("/predict", response_model=Prediction)
def predict(payload: TextIn):
try:
language = predict_language(payload.text)
return {"language": language}
except Exception as e:
raise HTTPException(status_code=400, detail=str(e))
|