Update app.py

app.py

@@ -424,8 +424,7 @@ with gr.Blocks(css="style.css") as demo:
                   
                     with gr.Row():
                         invert_button = gr.Button("⬆️ Invert")
-                    with gr.Row():
-                        file_output = gr.File(label="Download Sampled/Inverted Model", container=True, interactive=False)
+
 
 
                 with gr.Column():
@@ -475,29 +474,24 @@ with gr.Blocks(css="style.css") as demo:
 
     gr.Markdown("""<div style="text-align: justify;"> After sampling a new model or inverting, you can download the model below.""")
 
-    #with gr.Row():
-    #    file_output = gr.File(label="Download Sampled/Inverted Model", container=True, interactive=False)
+    with gr.Row():
+        file_output = gr.File(label="Download Sampled/Inverted Model", container=True, interactive=False)
         
 
 
 
 
-    invert_button.click(fn=run_inversion,
+        invert_button.click(fn=run_inversion,
                     inputs=[input_image, pcs, epochs, weight_decay,lr], 
                     outputs = [gallery, file_output])
         
             
-    sample.click(fn=sample_then_run, outputs=[gallery, file_output])
+        sample.click(fn=sample_then_run, outputs=[gallery, file_output])
 
         
-    submit.click(
+        submit.click(
         fn=edit_inference, inputs=[prompt, negative_prompt, cfg, steps, seed, injection_step, a1, a2, a3, a4], outputs=[gallery]
     )
         file_input.change(fn=file_upload, inputs=file_input, outputs = input_image)
 
-
-
-
-    
-    
-demo.queue().launch()
+demo.queue().launch()