Jakaria commited on
Commit
b89bd7e
·
1 Parent(s): 1245461

Add Bangla model API

Browse files
Files changed (1) hide show
  1. app.py +13 -3
app.py CHANGED
@@ -1,12 +1,22 @@
1
  from fastapi import FastAPI
 
2
 
3
  app = FastAPI()
4
 
 
 
 
 
 
 
5
  @app.get("/")
6
- def read_root():
7
  return {"message": "Bangla model API is running!"}
8
 
 
9
  @app.post("/predict")
10
  def predict(text: str):
11
- # Dummy response (replace with your ML model prediction)
12
- return {"input": text, "prediction": "This is a dummy result"}
 
 
 
1
  from fastapi import FastAPI
2
+ import joblib
3
 
4
  app = FastAPI()
5
 
6
+ # Load your model, vectorizer, label encoder
7
+ model = joblib.load("bangla_model.pkl")
8
+ vectorizer = joblib.load("bangla_vectorizer.pkl")
9
+ label_encoder = joblib.load("bangla_label_encoder.pkl")
10
+
11
+ # Root route to avoid 404
12
  @app.get("/")
13
+ def root():
14
  return {"message": "Bangla model API is running!"}
15
 
16
+ # Example predict endpoint
17
  @app.post("/predict")
18
  def predict(text: str):
19
+ vect = vectorizer.transform([text])
20
+ pred = model.predict(vect)
21
+ label = label_encoder.inverse_transform(pred)
22
+ return {"prediction": label[0]}