Update app.py

app.py

@@ -24,10 +24,10 @@ import matplotlib.pyplot as plt
   
 from transformers import BertForSequenceClassification, BertTokenizer
 
-model = BertForSequenceClassification.from_pretrained("RuudVelo/dutch_news_classifier_bert_finetuned")
+model = BertForSequenceClassification.from_pretrained("RuudVelo/dutch_news_clf_bert_finetuned")
 #from transformers import BertTokenizer
 
-tokenizer = BertTokenizer.from_pretrained("RuudVelo/dutch_news_classifier_bert_finetuned")
+tokenizer = BertTokenizer.from_pretrained("RuudVelo/dutch_news_clf_bert_finetuned")
 
 #text = ["this is one sentence", "this is another sentence"]
 text = st.text_area('Please type/copy/paste the Dutch article')
@@ -40,12 +40,13 @@ if text:
    
    fig = plt.figure()
    ax = fig.add_axes([0,0,1,1])
-   langs = ['Binnenland', 'Buitenland' ,'Cultuur & Media' ,'Economie' ,'Koningshuis',
+   labels_plot = ['Binnenland', 'Buitenland' ,'Cultuur & Media' ,'Economie' ,'Koningshuis',
  'Opmerkelijk' ,'Politiek', 'Regionaal nieuws', 'Tech']
-   students = probabilities[0].cpu().detach().numpy()
+   probs_plot = probabilities[0].cpu().detach().numpy()
    
-   ax.barh(langs,students)
+   ax.barh(labels_plot,probs_plot )
    st.pyplot(fig)
+   st.json(predictions)
    #plt.show()
    #out = pipe(text)
    #st.json(predictions)