Update app.py

app.py

@@ -228,6 +228,8 @@ with gr.Blocks(css=css) as demo:
                 )
                 generate_bt = gr.Button("Generate", scale=1)
 
+                gr.Button("Download", link="/file=all_files.zip")
+    
         # Image Result from last prompt
         image = gr.Image(type="filepath")
 
@@ -240,8 +242,9 @@ with gr.Blocks(css=css) as demo:
                 placeholder="Enter your prompt",
             )
             btn = gr.Button("Generate Gallery of Saved Images")
+            
         gallery = gr.Gallery(
-            label="Generated Images", show_label=False, elem_id="gallery"
+            label="Generated Images", show_label=True, elem_id="gallery"
         )
 
         with gr.Row(variant="compact"):
@@ -302,4 +305,4 @@ with gr.Blocks(css=css) as demo:
         clear_all_button.click(clear_all_button_click)
 
 demo.queue()
-demo.launch()
+demo.launch(allowed_paths=["/"])