Update app.py

app.py

@@ -89,10 +89,6 @@ def save_all_button_click():
 def clear_all_button_click():
     clear_all_images()
 
-# Attach click event handlers to the buttons
-save_all_button.click(save_all_button_click)
-clear_all_button.click(clear_all_button_click)
-
 # Add buttons to the Streamlit app
 gr.button(save_all_button)
 gr.button(clear_all_button)
@@ -208,6 +204,11 @@ css = """
 }
 """
 with gr.Blocks(css=css) as demo:
+    
+    # Attach click event handlers to the buttons
+    save_all_button.click(save_all_button_click)
+    clear_all_button.click(clear_all_button_click)
+
     with gr.Column(elem_id="container"):
         gr.Markdown(
             """##🕹️ Real Time 🎨 ImageGen Gallery 🌐""",