Add Bangla model API

app.py

@@ -1,22 +1,25 @@
 from fastapi import FastAPI
+from pydantic import BaseModel
 import joblib
 
 app = FastAPI()
 
-# Load your model, vectorizer, label encoder
+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")
 
-# Root route to avoid 404
 @app.get("/")
 def root():
     return {"message": "Bangla model API is running!"}
 
-# Example predict endpoint
 @app.post("/predict")
-def predict(text: str):
-    vect = vectorizer.transform([text])
+def predict(request: PredictRequest):
+    vect = vectorizer.transform([request.text])
     pred = model.predict(vect)
     label = label_encoder.inverse_transform(pred)
-    return {"prediction": label[0]}
+    # Convert numpy type to Python native type
+    return {"prediction": str(label[0])}