with Download Button

app.py

@@ -405,8 +405,8 @@ for ex_human in human_list_path:
 ##default human
 
 css = '''
-/*.gradio-container {width: 85% !important;
-.gradio-container {width: 100% !important; */
+/*.gradio-container {width: 85% !important; */
+.gradio-container {width: 100% !important; 
 background: linear-gradient(135deg, #CCFED9, #95BAFF);
 }
 #huggingface-space-header {
@@ -420,6 +420,7 @@ background: linear-gradient(135deg, #CCFED9, #95BAFF);
     color: white !important;
     }
 
+*/
 html, body {
     height: 100%;
     margin: 0;
@@ -430,7 +431,7 @@ html, body {
     
     background: transparent !important;
 }
-
+*/
 
 html body h1#app-title {
     text-align: center;
@@ -485,6 +486,10 @@ a[href*="api"], .api-button, #api-button {
     color: white !important;
     width: 100%;
 }
+#download_btn {
+background: #553566 !important;
+color: white !important;
+}
 input[type="radio"]:checked + span {
     border-color: #D46BCA !important; 
     }
@@ -522,16 +527,17 @@ with image_blocks as demo:
 
         with gr.Column():
             # 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)
+            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)
-            download_btn = gr.DownloadButton(label="Download Generated Image")
+        # 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():
         try_button = gr.Button(value="Try-on",  elem_id="generate_btn")
+        download_btn = gr.DownloadButton(label="Download Generated Image", elem_id="download_btn")
         with gr.Accordion(label="Advanced Settings", open=False):
             with gr.Row():
                 denoise_steps = gr.Number(label="Denoising Steps", minimum=20, maximum=40, value=30, step=1)