Update app.py

app.py

@@ -418,18 +418,14 @@ def file_upload(file):
     
     
     
-    
-    # <h1 style="margin-left: 12px;text-align: center;margin-bottom: 7px;display: inline-block">weights2weights</h1>
-    # <h3 style="display: inline-block;margin-left: 10px;margin-top: 6px;font-weight: 500">Interpreting the Weight Space of Customized Diffusion Models</h3>
-   
-
 intro = """
 <div style="display: flex;align-items: center;justify-content: center">
-    <h3 style="display: inline-block;margin-left: 10px;margin-top: 6px;font-weight: 500">Interpreting the Weight Space of Customized Diffusion Models</h3>
+    <h2 style="display: inline-block;margin-left: 10px;margin-top: 6px;font-weight: 500">Interpreting the Weight Space of Customized Diffusion Models (aka <b> <em>weights2weights</em></b>)</h2>
 </div>
 <p style="font-size: 0.95rem;margin: 0rem;line-height: 1.2em;margin-top:1em;display: inline-block">
-    <a href="https://snap-research.github.io/weights2weights/" target="_blank">project page</a> | <a href="https://arxiv.org/abs/2406.09413" target="_blank">paper</a>
+    <a href="https://snap-research.github.io/weights2weights/" target="_blank">Project Page</a> | <a href="https://arxiv.org/abs/2406.09413" target="_blank">Paper</a>
      | 
+     <a href="https://github.com/snap-research/weights2weights" target="_blank">Code</a> |
     <a href="https://huggingface.co/spaces/Snapchat/w2w-demo?duplicate=true" target="_blank" style="
         display: inline-block;
     ">
@@ -443,9 +439,7 @@ with gr.Blocks(css="style.css") as demo:
     gr.HTML(intro)
     with gr.Tab("Model Editing"):
         gr.Markdown("""
-        Click sample (to sample an identity) *or* upload an image & click `invert` to get started ✨
-        > 💡 When inverting, draw a mask over the face for improved results. 
-        > To use a model previously downloaded from this demo see `Uplaoding a model` in the `Advanced options`
+        Click the `Sample New Model` to sample a new identity-encoding model or upload a model to get started ✨
         """)
         with gr.Column():
                 with gr.Row():
@@ -456,7 +450,7 @@ with gr.Blocks(css="style.css") as demo:
 
 
                     with gr.Column():
-                        image_slider1 = ImageSlider(position=0.5, type="pil", height=512, width=512)
+                        image_slider1 = ImageSlider(position=0.5, type="pil", height=512, width=512, label= "Reference Identity | Generated Samples by User")
                         
                         prompt1 = gr.Textbox(label="Prompt",
                                                         info="Make sure to include 'sks person'" ,
@@ -488,6 +482,9 @@ with gr.Blocks(css="style.css") as demo:
                         submit1 = gr.Button("Generate")
 
     with gr.Tab("Inversion"):
+        gr.Markdown("""
+        Upload an image and optionally define a mask by drawing over the face. Then click `invert` to get started ✨
+        """)
         with gr.Row():
             with gr.Column():
                 input_image = gr.ImageEditor(elem_id="image_upload", type='pil', label="Upload image and draw to define mask", height=512, width=512, brush=gr.Brush(), layers=False) 
@@ -499,7 +496,7 @@ with gr.Blocks(css="style.css") as demo:
                 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)
+                image_slider2 = ImageSlider(position=0.5, type="pil", height=512, width=512, label= "Reference Identity | Generated Samples by User")
                             
                 prompt2 = gr.Textbox(label="Prompt",
                                     info="Make sure to include 'sks person'" ,
@@ -529,9 +526,8 @@ with gr.Blocks(css="style.css") as demo:
                         
 
                 submit2 = gr.Button("Generate")
-                
-
 
+        
 
         
                     
@@ -549,5 +545,9 @@ with gr.Blocks(css="style.css") as demo:
     submit2.click(fn=edit_inference, inputs=[ prompt2, negative_prompt2, cfg2, steps2, seed2, injection_step2, a1_2, a2_2, a3_2, a4_2], outputs=image_slider2)
 
 
+  
+
+
+
  
 demo.queue().launch()