Hugging Face's logo

Spaces:
jflo
/
text_classification
Running

App Files Community
jflo commited on
Commit
0f6d102
·
1 Parent(s): eb09543

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -30,7 +30,6 @@ def sentiment_analysis(payload: ClassificationRequest):
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],
35
  category_list[1]: pred_prob[1],
36
  category_list[2]: pred_prob[2],
 
30
  pred_prob = model.predict_proba(X)
31
 
32
  return {
 
33
  category_list[0]: pred_prob[0],
34
  category_list[1]: pred_prob[1],
35
  category_list[2]: pred_prob[2],