Jakaria commited on
Commit
29e19f2
·
1 Parent(s): a5749f2

Add Bangla model API

Browse files
Files changed (1) hide show
  1. app.py +5 -1
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
- return {"prediction": str(label[0])}
 
 
 
 
 
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
+