fixed the return with probs

app.py

@@ -13,9 +13,9 @@ def predict(news_headline):
 
     probabilities = learner.predict(news_headline)
 
-    values, indices = torch.topk(probabilities[0]['probs'], 5)
+    # values, indices = torch.topk(probabilities[0]['probs'], 5)
 
-    return {LABELS[i]: v.item() for i, v in zip(indices, values)}
+    return {LABELS[i]: probabilities[0]['probs'][i] for i, v in range(len(LABELS))}
 
 interface = gr.Interface(
     predict,