Update app.py

app.py

@@ -67,8 +67,9 @@ if st.button('Submit'):
         probs_plot = probabilities[0].cpu().detach().numpy()
    
         ax.barh(labels_plot,probs_plot)
-        ax.set_xlabel("Predicted category")
-        ax.set_ylabel("Probability")
+        ax.set_title("Probability")
+        ax.set_xlabel("Probability")
+        ax.set_ylabel("Predicted category")
         st.pyplot(fig)
    #     output = genQuestion(option, input)
    #     print(output)