Update app.py

app.py

@@ -35,9 +35,10 @@ tokenizer = BertTokenizer.from_pretrained("RuudVelo/dutch_news_clf_bert_finetune
 # Title
 st.title("Dutch news article classification")
 
-#image = Image.open('dataset-cover_articles.jpg')
-st.image('dataset-cover_articles.jpeg', width=100)
+st.write("This app classifies a Dutch news article into one of 9 pre-defined* article categories")
 
+#image = Image.open('dataset-cover_articles.jpg')
+st.image('dataset-cover_articles.jpeg', width=150)
 
 text = st.text_area('Please type/copy/paste text of the Dutch article')
 
@@ -82,6 +83,7 @@ if st.button('Submit'):
    # st.write(output)   
 #encoding = tokenizer(text, return_tensors="pt")
 #import numpy as np
+st.write("* The predefined categories are Binnenland, Buitenland, Cultuur & Media, Economie , Koningshuis, Opmerkelijk, Politiek, 'Regionaal nieuws en Tech")
 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")
 st.write('\n')
 st.write('The model performance details can be found at https://huggingface.co/RuudVelo/dutch_news_classifier_bert_finetuned')