import streamlit as st | |
from transformers import pipeline | |
pipe = pipeline(model="RuudVelo/dutch_news_classifier_bert_finetuned") | |
text = st.text_area('Please type/copy/paste the Dutch article') | |
labels = ['Binnenland' 'Buitenland' 'Cultuur & Media' 'Economie' 'Koningshuis' | |
'Opmerkelijk' 'Politiek' 'Regionaal nieuws' 'Tech'] | |
if text: | |
out = pipe(text) | |
st.json(out) |