import streamlit as st
from transformers import pipeline

# Load the text generation model
generator = pipeline("text-generation", model="microsoft/phi-1_5")

# Streamlit app title
st.title("Text Generation with DistilGPT-2")

# Input text from the user
input_text = st.text_area("Enter your prompt:", "In this course, we will teach you how to")

# Number of sequences to generate
num_sequences = st.slider("Number of sequences to generate:", 1, 5, 2)

# Generate button
if st.button("Generate Text"):
    with st.spinner("Generating..."):
        # Generate text using the model
        generated_texts = generator(input_text, max_length=30)
    
    # Display generated texts
    for i, generated in enumerate(generated_texts):
        st.subheader(f"Generated Text {i + 1}")
        st.write(generated['generated_text'])