| 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, max_length=40) | |
| st.write(output[0]['generated_text']) |