Update app.py

app.py

@@ -69,14 +69,14 @@ if st.button('Submit'):
         number = predictions[0].cpu().detach().numpy()
         probabilities = torch.nn.functional.softmax(outputs.logits, dim=-1)
    
-        fig = plt.figure(figsize=(10,5))
+        fig = plt.figure(figsize=(10,4))
         ax = fig.add_axes([0,0,1,1])
         labels_plot = ['Binnenland', 'Buitenland' ,'Cultuur & Media' ,'Economie' ,'Koningshuis',
  'Opmerkelijk' ,'Politiek', 'Regionaal nieuws', 'Tech']
         probs_plot = probabilities[0].cpu().detach().numpy()*100
    
         ax.barh(labels_plot,probs_plot)
-        ax.set_title("Predicted article category probability")
+        ax.set_title("Predicted article category probability", fontsize=20)
         ax.set_xlabel("Probability")
         ax.set_ylabel("Predicted category")
         st.pyplot(fig)