RuudVelo commited on
Commit
a5f0a48
1 Parent(s): 5772d0d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -79,6 +79,11 @@ if st.button('Submit'):
79
  ax.set_title("Predicted article category probability", fontsize=20)
80
  ax.set_xlabel("Probability (%)", fontsize=16)
81
  ax.set_ylabel("Predicted category", fontsize=16)
 
 
 
 
 
82
  st.pyplot(fig)
83
 
84
  st.write('The predicted category is: **{}** with a probability of: **{:.1f}%**'.format(labels_plot[number],(probs_plot[predictions])*1))
 
79
  ax.set_title("Predicted article category probability", fontsize=20)
80
  ax.set_xlabel("Probability (%)", fontsize=16)
81
  ax.set_ylabel("Predicted category", fontsize=16)
82
+
83
+ # change the fontsize
84
+ ax.set_xticklabels(probs_plot, fontsize=14)
85
+ ax.set_yticklabels(labels_plot, fontsize=14)
86
+
87
  st.pyplot(fig)
88
 
89
  st.write('The predicted category is: **{}** with a probability of: **{:.1f}%**'.format(labels_plot[number],(probs_plot[predictions])*1))