RuudVelo commited on
Commit
b8a98b8
·
1 Parent(s): 5cd0d07

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -81,13 +81,13 @@ if st.button('Submit'):
81
  ax.set_ylabel("Predicted category")
82
  st.pyplot(fig)
83
 
84
- st.write('The predicted category is: {} with a probability of: {:.1f}%'.format(labels_plot[number],(probs_plot[predictions])*1))
85
  # output = genQuestion(option, input)
86
  # print(output)
87
  # st.write(output)
88
  #encoding = tokenizer(text, return_tensors="pt")
89
  #import numpy as np
90
- st.write("* The predefined categories are Binnenland, Buitenland, Cultuur & Media, Economie , Koningshuis, Opmerkelijk, Politiek, 'Regionaal nieuws en Tech")
91
  st.write("The model for this app has been trained using data from Dutch news articles published by NOS. For more information regarding the dataset can be found at https://www.kaggle.com/maxscheijen/dutch-news-articles")
92
  #st.write('\n')
93
  st.write('The model performance details can be found at https://huggingface.co/RuudVelo/dutch_news_classifier_bert_finetuned')
 
81
  ax.set_ylabel("Predicted category")
82
  st.pyplot(fig)
83
 
84
+ st.write('The predicted category is: **{}** with a probability of: {:.1f}%'.format(labels_plot[number],(probs_plot[predictions])*1))
85
  # output = genQuestion(option, input)
86
  # print(output)
87
  # st.write(output)
88
  #encoding = tokenizer(text, return_tensors="pt")
89
  #import numpy as np
90
+ st.write("The pre-defined categories are Binnenland, Buitenland, Cultuur & Media, Economie , Koningshuis, Opmerkelijk, Politiek, 'Regionaal nieuws en Tech")
91
  st.write("The model for this app has been trained using data from Dutch news articles published by NOS. For more information regarding the dataset can be found at https://www.kaggle.com/maxscheijen/dutch-news-articles")
92
  #st.write('\n')
93
  st.write('The model performance details can be found at https://huggingface.co/RuudVelo/dutch_news_classifier_bert_finetuned')