RuudVelo's picture
Update app.py
093cd61
raw
history blame
371 Bytes
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)