Spaces:

RuudVelo
/

Dutch_news_classifier

Runtime error

RuudVelo commited on Mar 24, 2022

Commit

dede7cb

1 Parent(s): b28d4fd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

  'Opmerkelijk' ,'Politiek', 'Regionaal nieuws', 'Tech']
         probs_plot = probabilities[0].cpu().detach().numpy()
+        ax.barh(labels_plot,probs_plot,title="Probability of article category" )
         st.pyplot(fig)
    #     output = genQuestion(option, input)
    #     print(output)