import streamlit as st

st.title("Image Generation")


prompt = st.text_input("Enter a prompt to generate an image:", value="pipeline under the sea")

if st.button("Generate Image"):
    with st.spinner("Generating image..."):
        image = pipe(prompt).images[0]
        st.image(image, caption="Generated Image", use_column_width=True)