Update app.py

app.py

@@ -1,4 +1,11 @@
 import streamlit as st
 
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
+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)