import streamlit as st
from transformers import pipeline

with st.spinner('Loading TaatikNet framework...'):
    pipe = pipeline("text2text-generation", model='malper/taatiknet', device_map="auto")

st.success('Loaded!')

text = st.text_area('Enter text and press ctrl/command+enter:')

if text:
    output = pipe(text)
    st.write(output[0]['generated_text'])