Add score to results

app.py

@@ -5,7 +5,7 @@ pipeline = pipeline(task="text-classification", model="databoyface/distilbert-ba
 
 def predict(input_txt):
     predictions = pipeline(input_txt)
-    return {p["label"] for p in predictions}
+    return {p["label"]: p["score"] for p in predictions}
     
 gradio_app = gr.Interface(
     fn=predict,