Hugging Face's logo

Spaces:
jflo
/
text_classification
Running

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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -27,8 +27,10 @@ def sentiment_analysis(payload: ClassificationRequest):
27
  category_list = ["Politics", "Sport", "Technology", "Entertainment", "Business"]
28
 
29
  predictions = model.predict(X)
 
30
  pred_prob = model.predict_proba(X)
31
-
 
32
  return {
33
  category_list[0]: pred_prob[0],
34
  category_list[1]: pred_prob[1],
 
27
  category_list = ["Politics", "Sport", "Technology", "Entertainment", "Business"]
28
 
29
  predictions = model.predict(X)
30
+
31
  pred_prob = model.predict_proba(X)
32
+ pred_prob = pred_prob.tolist()[0]
33
+
34
  return {
35
  category_list[0]: pred_prob[0],
36
  category_list[1]: pred_prob[1],