ilan541 commited on
Commit
99116b6
·
1 Parent(s): 63b3859

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -42,7 +42,7 @@ def predict(your_text):
42
 
43
  # print predicted class
44
  i = np.argmax(y_pred_logits)
45
- return f"Predicted class: {labels[i]} with proba {y_probs[i]:.4f}"
46
 
47
  iface = gr.Interface(fn=predict, inputs="text", outputs="text")
48
  iface.launch()
 
42
 
43
  # print predicted class
44
  i = np.argmax(y_pred_logits)
45
+ return f"Predicted class: {labels[i]} with proba {y_probs[i]:.4f}\n{labels}{y_probs}"
46
 
47
  iface = gr.Interface(fn=predict, inputs="text", outputs="text")
48
  iface.launch()