Update app.py

app.py

@@ -388,9 +388,6 @@ def file_upload(file):
     
 
     
-    
-    
-    
 intro = """
 <div style="display: flex;align-items: center;justify-content: center">
     <h1 style="margin-left: 12px;text-align: center;margin-bottom: 7px;display: inline-block">weights2weights</h1>
@@ -411,16 +408,25 @@ intro = """
 with gr.Blocks(css="style.css") as demo:
     gr.HTML(intro)
     
-    gr.Markdown("""<div style="text-align: justify;"> Click below to sample an identity-encoding model, or upload an image below and click \"invert\". You can also optionally draw over the face to define a mask. To use model previously downloaded from this demo see \"Uplaoding a model\" in the Advanced options""")
+    gr.Markdown("""<div style="text-align: justify;"> In this demo, you can get an identity-encoding model by sampling or inverting. To use a model previously downloaded from this demo see \"Uploading a model\" in the Advanced options. Next, you can generate new samples from it, or edit the identity encoded in the model and generate samples from the edited model. We provide detailed instructions and tips at the bottom of the page.""")
     with gr.Column():
             with gr.Row():
                 with gr.Column():
+                    with gr.Row():
+                        gr.Markdown("""Click below to sample a new identity-encoding model.""")
+
+                    with gr.Row():
+                        sample = gr.Button("🎲 Sample New Model")
+                    
+                    gr.Markdown("""Or upload an image and optionally draw over the face to define a mask. Then click `invert`. """)
                     input_image = gr.Image(source='upload', elem_id="image_upload", tool='sketch', type='pil', label="Upload image and draw to define mask",
                                             height=512, width=512, brush_color='#00FFFF', mask_opacity=0.6)
                   
                     with gr.Row():
-                        sample = gr.Button("🎲 Sample New Model")
                         invert_button = gr.Button("⬆️ Invert")
+
+
+
                 with gr.Column():
                     gallery = gr.Image(label="Image",height=512, width=512, interactive=False)
                     
@@ -469,24 +475,33 @@ 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 Model", container=True, interactive=False)
+        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])
-    submit.click(
+                    outputs = [gallery, file_output])
+        
+            
+        sample.click(fn=sample_then_run, outputs=[gallery, file_output])
+
+        
+        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)
+        file_input.change(fn=file_upload, inputs=file_input, outputs = input_image)
+
+
 
 
     
+    
+    
 
+demo.queue().launch(share=True)