Fahad Ebrahim commited on
Commit
fa7e059
·
1 Parent(s): a2af74c

Dict output

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ def text_classification(text):
23
  preprocessed_issue = preprocess (text)
24
  out = classifier(preprocessed_issue)[0]
25
  label_scores = {result['label']: result['score'] for result in out}
26
- return out
27
 
28
  examples=["This is a question", "This is a bug", "This is an enhancement" ]
29
 
 
23
  preprocessed_issue = preprocess (text)
24
  out = classifier(preprocessed_issue)[0]
25
  label_scores = {result['label']: result['score'] for result in out}
26
+ return label_scores
27
 
28
  examples=["This is a question", "This is a bug", "This is an enhancement" ]
29