File size: 659 Bytes
34e5a0f a5749f2 b89bd7e ba47e53 613cb3e a5749f2 34e5a0f b89bd7e 613cb3e b89bd7e 34e5a0f 298ba53 ba47e53 a5749f2 34e5a0f a5749f2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | from fastapi import FastAPI
from pydantic import BaseModel
import joblib
app = FastAPI()
class PredictRequest(BaseModel):
text: str
# Load models
model = joblib.load("bangla_model.pkl")
vectorizer = joblib.load("bangla_vectorizer.pkl")
label_encoder = joblib.load("bangla_label_encoder.pkl")
@app.get("/")
def root():
return {"message": "Bangla model API is running!"}
@app.post("/predict")
def predict(request: PredictRequest):
vect = vectorizer.transform([request.text])
pred = model.predict(vect)
label = label_encoder.inverse_transform(pred)
# Convert numpy type to Python native type
return {"prediction": str(label[0])}
|