Update app.py

app.py

@@ -66,7 +66,9 @@ if st.button('Submit'):
  'Opmerkelijk' ,'Politiek', 'Regionaal nieuws', 'Tech']
         probs_plot = probabilities[0].cpu().detach().numpy()
    
-        ax.barh(labels_plot,probs_plot,title="Probability of article category" )
+        ax.barh(labels_plot,probs_plot)
+        ax.set_xlabel("Predicted category")
+        ax.set_ylabel("Probability")
         st.pyplot(fig)
    #     output = genQuestion(option, input)
    #     print(output)