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 |
+
|