Update app.py

app.py

@@ -69,7 +69,7 @@ if st.button('Submit'):
         number = predictions[0].cpu().detach().numpy()
         probabilities = torch.nn.functional.softmax(outputs.logits, dim=-1)
    
-        fig = plt.figure()
+        fig = plt.figure(figsize=(1,1))
         ax = fig.add_axes([0,0,1,1])
         labels_plot = ['Binnenland', 'Buitenland' ,'Cultuur & Media' ,'Economie' ,'Koningshuis',
  'Opmerkelijk' ,'Politiek', 'Regionaal nieuws', 'Tech']
@@ -81,7 +81,7 @@ if st.button('Submit'):
         ax.set_ylabel("Predicted category")
         st.pyplot(fig)
         
-        st.write('The predicted category is: **{}** with a probability of: {:.1f}%'.format(labels_plot[number],(probs_plot[predictions])*1))
+        st.write('The predicted category is: **{}** with a probability of: **{:.1f}%**'.format(labels_plot[number],(probs_plot[predictions])*1))
    #     output = genQuestion(option, input)
    #     print(output)
    # st.write(output)