Jakaria
commited on
Commit
·
29e19f2
1
Parent(s):
a5749f2
Add Bangla model API
Browse files
app.py
CHANGED
|
@@ -22,4 +22,8 @@ def predict(request: PredictRequest):
|
|
| 22 |
pred = model.predict(vect)
|
| 23 |
label = label_encoder.inverse_transform(pred)
|
| 24 |
# Convert numpy type to Python native type
|
| 25 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 22 |
pred = model.predict(vect)
|
| 23 |
label = label_encoder.inverse_transform(pred)
|
| 24 |
# Convert numpy type to Python native type
|
| 25 |
+
if label[0]==1:
|
| 26 |
+
return {"prediction": "Hate"}
|
| 27 |
+
else:
|
| 28 |
+
return {"prediction": "Not Hate"}
|
| 29 |
+
|