Update app.py

app.py

@@ -133,10 +133,10 @@ def main():
 
     # Show download and reset buttons only if an image is generated
     if st.session_state["image"]:
+        # Download button
         image_bytes = io.BytesIO()
         st.session_state["image"].save(image_bytes, format='PNG')
         
-        # Download button
         st.download_button(
             label="Download Image",
             data=image_bytes.getvalue(),