refactor results

app.py

@@ -10,11 +10,15 @@ model = AutoModelForSequenceClassification.from_pretrained("serdarakyol/interpre
 
 def prediction(news):
     # create pipeline
-    clasifer = pipeline("sentiment-analysis", tokenizer=tokenizer, model=model)
+    clasifer = pipeline("sentiment-analysis", tokenizer=tokenizer, model=model, return_all_scores=True)
 
-    pred = clasifer(news)
-    return str(pred[0]["label"]), str(round(pred[0]["score"], 5))
+    preds = clasifer(news)
 
+    preds_dict={}
+    for pred in preds[0]:
+        preds_dict[pred['label']] = pred['score']
+
+    return preds_dict
 
 
 gradio_ui = gr.Interface(
@@ -27,11 +31,8 @@ gradio_ui = gr.Interface(
         ["Instagram, 3 yıl önce kullanıcıların beğenisine sunduğu IGTV özelliğini kaldırıyor. IGTV’nin yerine gelecek yeni özelliğin ise platformun Reels'tan elde ettiği başarıyı pekiştirip içerik üreticileri için kullanışlı bir zemin oluşturması bekleniyor."],
         ["Kamu bankaları Halkbank, Vakıfbank ve Ziraat Bankası ortak bir açıklama yayınlayarak kredi faiz oranlarında indirim yapıldığını açıkladı. Söz konusu indirimler ticari ve konut kredileri kapsıyor. Tüketici kredisi kapsam dışında bırakıldı."]
     ],
-    inputs=gr.inputs.Textbox(lines=10, label="Paste some text here"),
-    outputs=[
-        gr.outputs.Textbox(label="News Category Label"),
-        gr.outputs.Textbox(label="News Category Score"),
-    ],
+    inputs=gr.inputs.Textbox(lines=10, label="Paste some news here"),
+    outputs=gr.outputs.Label(num_top_classes=5, type="auto", label="News categories probabilities"),
     theme="huggingface",
     article="<p style='text-align: center'>for our other AI works: <a href='https://www.kodiks.com/ai_solutions.html' target='_blank'>https://www.kodiks.com/ai_solutions.html</a> | <a href='https://twitter.com/KodiksBilisim' target='_blank'>Contact us</a></p>",
 )