Update app.py

app.py

@@ -67,7 +67,7 @@ if st.button('Submit'):
         probs_plot = probabilities[0].cpu().detach().numpy()
    
         ax.barh(labels_plot,probs_plot)
-        ax.set_title("Probability")
+        ax.set_title("Predicted article category probability")
         ax.set_xlabel("Probability")
         ax.set_ylabel("Predicted category")
         st.pyplot(fig)