jflo commited on
Commit
eb09543
·
1 Parent(s): 1b0a68c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -29,7 +29,6 @@ def sentiment_analysis(payload: ClassificationRequest):
29
  predictions = model.predict(X)
30
  pred_prob = model.predict_proba(X)
31
 
32
-
33
  return {
34
  category_list[predictions.tolist()[0]]: pred_prob[predictions.tolist()[0]],
35
  category_list[0]: pred_prob[0],
@@ -37,5 +36,5 @@ def sentiment_analysis(payload: ClassificationRequest):
37
  category_list[2]: pred_prob[2],
38
  category_list[3]: pred_prob[3],
39
  category_list[4]: pred_prob[4]
40
- }
41
 
 
29
  predictions = model.predict(X)
30
  pred_prob = model.predict_proba(X)
31
 
 
32
  return {
33
  category_list[predictions.tolist()[0]]: pred_prob[predictions.tolist()[0]],
34
  category_list[0]: pred_prob[0],
 
36
  category_list[2]: pred_prob[2],
37
  category_list[3]: pred_prob[3],
38
  category_list[4]: pred_prob[4]
39
+ }
40