import streamlit as st
from transformers import pipeline

# Load a pre-trained text generation model from Hugging Face
model = pipeline("text-generation", model="gpt2")

# Streamlit UI components
st.title("Text Generation with Streamlit")
prompt = st.text_input("Enter your prompt:")

if st.button("Generate"):
    # Generate text using the model
    generated_text = model(prompt, max_length=1000, num_return_sequences=1)[0]["generated_text"]
    st.write("Generated Text:")
    st.write(generated_text)