import streamlit as st
from transformers import pipeline, set_seed

# Define pipeline for text generation
generator = pipeline('text-generation', model='gpt2')

# Set seed for reproducibility
set_seed(42)

# Set up Streamlit app
st.title('Text Generation with Hugging Face')
st.write('Type a prompt below and watch the model generate text!')

# Get user input
prompt = st.text_input('Enter your prompt here:')

# Generate text
if prompt:
    text = generator(prompt, max_length=50, num_return_sequences=1)[0]['generated_text']
    st.write(text)