Update app.py

app.py

@@ -425,6 +425,8 @@ with gr.Blocks(css="style.css") as demo:
                 with gr.Column():
                     gr.Markdown("""2) Generate images of the sampled/inverted identity or edit the identity and generate new images. """)
                     gallery = gr.Image(label="Image",height=512, width=512, interactive=False)
+                    submit = gr.Button("Generate")
+
                     
             prompt = gr.Textbox(label="Prompt",
                                             info="Make sure to include 'sks person'" ,
@@ -464,7 +466,6 @@ with gr.Blocks(css="style.css") as demo:
 
                     file_input = gr.File(label="Upload Model", container=True)
 
-            submit = gr.Button("Generate")