jflo commited on
Commit
1a41364
·
1 Parent(s): 46f51ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -28,5 +28,11 @@ def sentiment_analysis(payload: SentimentRequest):
28
  predictions = model.predict(X)
29
  pred_prob = model.predict_proba(X)
30
 
31
- return [predictions.tolist()[0],pred_prob.tolist()[0][0],pred_prob.tolist()[0][1]]
 
 
 
 
 
 
32
 
 
28
  predictions = model.predict(X)
29
  pred_prob = model.predict_proba(X)
30
 
31
+ pred_list = [predictions.tolist()[0],pred_prob.tolist()[0][0],pred_prob.tolist()[0][1]]
32
+
33
+ return {
34
+ "gender": pred_list[0],
35
+ "f_perc": pred_list[1],
36
+ "m_perc": pred_list[2]
37
+ }
38