Add Bangla model API

app.py

@@ -1,20 +1,22 @@
 from fastapi import FastAPI
-from pydantic import BaseModel
 import joblib
 
-# Load your trained model + vectorizer + encoder
+app = FastAPI()
+
+# Load your model, vectorizer, label encoder
 model = joblib.load("bangla_model.pkl")
 vectorizer = joblib.load("bangla_vectorizer.pkl")
 label_encoder = joblib.load("bangla_label_encoder.pkl")
 
-app = FastAPI()
-
-class TextData(BaseModel):
-    text: str
+# Root route to avoid 404
+@app.get("/")
+def root():
+    return {"message": "Bangla model API is running!"}
 
+# Example predict endpoint
 @app.post("/predict")
-def predict(data: TextData):
-    X = vectorizer.transform([data.text])
-    pred = model.predict(X)[0]
-    label = label_encoder.inverse_transform([pred])[0]
-    return {"prediction": label}
+def predict(text: str):
+    vect = vectorizer.transform([text])
+    pred = model.predict(vect)
+    label = label_encoder.inverse_transform(pred)
+    return {"prediction": label[0]}