Update app.py

app.py

@@ -136,8 +136,6 @@ def main():
         key="prompt_input"
     )
 
-    # Create a placeholder for the spinner
-    spinner_placeholder = st.empty()
 
     # Generate Image button
     generate_button_clicked = st.button("Generate Image")
@@ -149,6 +147,9 @@ def main():
         else:
             st.warning("Please enter a prompt.")
 
+    # Create a placeholder for the spinner
+    spinner_placeholder = st.empty()
+    
     # Show the image if it exists in the session state
     if st.session_state.get("image"):
         st.image(st.session_state["image"], caption="Generated Image")