File size: 489 Bytes
9cbeca2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import streamlit as st
import transformers


model = transformers.AutoModelForCausalLM.from_pretrained("distilgpt2")
tokenizer = transformers.AutoTokenizer.from_pretrained("distilgpt2")

user_inputed_text = st.text_input("Insert text")

outputs = model.generate(
    **tokenizer([user_inputed_text], return_tensors='pt'),
    max_new_tokens=50, do_sample=True,
)


outputs_text = tokenizer.decode(outputs[0])

st.text_area(label='output', value="your text is :" + user_inputed_text[::-1])