show TOP-3 predictions with proba

app.py

@@ -14,8 +14,11 @@ except Exception as e:
     print(f"ERROR: loading model failed with: {str(e)}")
 
 def classify(text):
-    pred_classes = model.predict([text])
-    output_text = ' '.join(pred_classes)
+    embed = model._texts2vecs([text])
+    probs = model.classifier.predict_proba(embed)
+    best_n = np.flip(np.argsort(probs, axis=1,)[0,-3:])
+    preds = [f"{model.classifier.classes_[i]} - {probs[0][i]*100:.1f}%" for i in best_n]
+    output_text = '\n'.join(preds)
     return output_text
 
 print('INFO: starting gradio interface')