Update app.py

app.py

@@ -24,5 +24,4 @@ def generate_image(input_text):
 
 if prompt:
     output = generate_image(prompt)
-    st.image(output, caption="Generated image")
-    st.download_button("Download", output)
+    st.image(output, caption="Generated image")