with Download Button

app.py

@@ -529,10 +529,10 @@ with image_blocks as demo:
             # image_out = gr.Image(label="Output", elem_id="output-img", height=400)
             masked_img = gr.Image(label="Masked image output", elem_id="masked-img",show_share_button=False, show_download_button=False)
        
-        # with gr.Column():
-        #     # image_out = gr.Image(label="Output", elem_id="output-img", height=400)
-        #     image_out = gr.Image(label="Output", elem_id="output-img",show_share_button=False, show_download_button=False)
-        #     download_btn = gr.DownloadButton(label="Download Generated Image", elem_id="download_btn")
+        with gr.Column():
+            # image_out = gr.Image(label="Output", elem_id="output-img", height=400)
+            image_out = gr.Image(label="Output", elem_id="output-img",show_share_button=False, show_download_button=False)
+            # download_btn = gr.DownloadButton(label="Download Generated Image", elem_id="download_btn")
 
 
     with gr.Column():