# Prompt:  Create a streamlit program that uses Text generation with a huggingface model
import streamlit as st
from transformers import pipeline

# Load the Hugging Face model for text generation
generator = pipeline("text-generation", model="gpt2")

# Set the maximum length of generated text
MAX_LENGTH = 100

# Create a Streamlit app
st.title("Text Generator")

# Add a text input for the user to provide a prompt for the generated text
prompt = st.text_input("Enter a prompt for the generated text:")

# Generate text based on the user's prompt
if prompt:
    generated_text = generator(prompt, max_length=MAX_LENGTH, do_sample=True)[0]["generated_text"]
    st.write(generated_text)