Update app.py

app.py

@@ -496,6 +496,7 @@ with gr.Blocks(css="style.css") as demo:
                 epochs = gr.Slider(label="Epochs", value=400, step=1, minimum=1, maximum=2000, interactive=True)
                 weight_decay = gr.Number(value=1e-10, label="Weight Decay", interactive=True)
                 invert_button = gr.Button("Invert")
+                file_output2 = gr.File(label="Download Inverted Model", container=True, interactive=False)
 
             with gr.Column():
                 image_slider2 = ImageSlider(position=0.5, type="pil", height=512, width=512)
@@ -528,7 +529,7 @@ with gr.Blocks(css="style.css") as demo:
                         
 
                 submit2 = gr.Button("Generate")
-                file_output2 = gr.File(label="Download Inverted Model", container=True, interactive=False)
+