import streamlit as st
from transformers import pipeline

# Load the text generation pipeline
generator = pipeline("text-generation")

# Streamlit app
def main():
    st.title("Text Generation App")

    # User input
    user_input = st.text_input("Enter a sentence:")

    if user_input:
        # Perform text generation
        result = generator(user_input)

        # Display result
        result_display = st.empty() 
        result_display.text_area("Generated text:", result[0]["generated_text"], height=171)

if __name__ == "__main__":
    main()