Dict output

app.py

@@ -22,6 +22,7 @@ def text_classification(text):
     classifier = TextClassificationPipeline(model=model, tokenizer=tokenizer, max_length=256, padding="max_length", truncation=True,top_k=None)
     preprocessed_issue = preprocess (text)
     out = classifier(preprocessed_issue)[0]
+    label_scores = {result['label']: result['score'] for result in out}
     return out
 
 examples=["This is a question", "This is a bug", "This is an enhancement" ]