Jakaria commited on
Commit
ba47e53
·
1 Parent(s): f333ee5

Add Bangla model API

Browse files
Dockerfile ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM python:3.9
2
+
3
+ WORKDIR /app
4
+
5
+ COPY requirements.txt .
6
+ RUN pip install --no-cache-dir -r requirements.txt
7
+
8
+ COPY . .
9
+
10
+ CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
app.py ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from fastapi import FastAPI
2
+ from pydantic import BaseModel
3
+ import joblib
4
+
5
+ # Load your trained model + vectorizer + encoder
6
+ model = joblib.load("bangla_model.pkl")
7
+ vectorizer = joblib.load("bangla_vectorizer.pkl")
8
+ label_encoder = joblib.load("bangla_label_encoder.pkl")
9
+
10
+ app = FastAPI()
11
+
12
+ class TextData(BaseModel):
13
+ text: str
14
+
15
+ @app.post("/predict")
16
+ def predict(data: TextData):
17
+ X = vectorizer.transform([data.text])
18
+ pred = model.predict(X)[0]
19
+ label = label_encoder.inverse_transform([pred])[0]
20
+ return {"prediction": label}
bangla_label_encoder.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8b97381ffedc27c901f65d34df96c460a6af7f866fd2a7866ee2c37f01545b34
3
+ size 343
bangla_model.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:588bea0a10ea919ecebaef1d252474b96eeb38a31ad06b7827512d9b3023ca12
3
+ size 2808971
bangla_vectorizer.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6aba876ea6ce6ffb627043624d1e9159beeaef30e2e0eea75d81986962dcb2f6
3
+ size 7530201
requirements.txt ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ fastapi
2
+ uvicorn
3
+ scikit-learn
4
+ joblib