import streamlit as st
from transformers import GPT2Tokenizer, GPT2Model

tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
model = GPT2Model.from_pretrained('gpt2')

st.title("GPT-2 Text Generation")

text = st.text_input("Enter the starting text:", "Replace me by any text you'd like.")

if st.button("Generate"):
    encoded_input = tokenizer(text, return_tensors='pt')
    output = model(**encoded_input)
    generated_text = tokenizer.decode(output[0][0][encoded_input[0][0].tolist().index(tokenizer.eos_token_id):])
    st.write(generated_text)