Update app.py

app.py

@@ -220,47 +220,66 @@ bags_min = torch.min(proj@bags[0]/(torch.norm(bags))**2).item()
 
 
 
-css = ''
-with gr.Blocks(css=css) as demo:
-    gr.Markdown("# <em>weights2weights</em> Demo")
-    gr.Markdown("Demo for the [h94/IP-Adapter-FaceID model](https://huggingface.co/h94/IP-Adapter-FaceID) - Generate AI images with your own face - Non-commercial license")
+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>
+    <h3 style="display: inline-block;margin-left: 10px;margin-top: 6px;font-weight: 500">Interpreting the Weight Space of Customized Diffusion Models</h3>
+</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://huggingface.co/spaces/Snapchat/w2w-demo?duplicate=true" target="_blank" style="
+        display: inline-block;
+    ">
+    <img style="margin-top: -1em;margin-bottom: 0em;position: absolute;" src="https://bit.ly/3CWLGkA" alt="Duplicate Space"></a>
+</p>
+"""
+
+with gr.Blocks(css="style.css") as demo:
+    gr.HTML(intro)
     with gr.Row():
         with gr.Column():
-            sample = gr.Button("Sample New Model")
             gallery1 = gr.Gallery(label="Identity from Sampled Model")
+            sample = gr.Button("Sample New Model")
+        gallery2 = gr.Gallery(label="Identity from Edited Model")
+            
 
+    with gr.Row():
         with gr.Column():          
-            prompt = gr.Textbox(label="Prompt",
-                        info="Make sure to include 'sks person'" ,
-                        placeholder="sks person", 
-                        value="sks person")
-            negative_prompt = gr.Textbox(label="Negative Prompt", placeholder="low quality, blurry, unfinished, cartoon", value="low quality, blurry, unfinished, cartoon")
-            seed = gr.Number(value=5, label="Seed", interactive=True)
-            cfg = gr.Slider(label="CFG", value=3.0, step=0.1, minimum=0, maximum=10, interactive=True)
-            steps = gr.Slider(label="Inference Steps",  value=50, step=1, minimum=0, maximum=100, interactive=True)
-            injection_step = gr.Slider(label="Injection Step",  value=800, step=1, minimum=0, maximum=1000, interactive=True)
-
+                prompt = gr.Textbox(label="Prompt",
+                            info="Make sure to include 'sks person'" ,
+                            placeholder="sks person", 
+                            value="sks person")
+                negative_prompt = gr.Textbox(label="Negative Prompt", placeholder="low quality, blurry, unfinished, cartoon", value="low quality, blurry, unfinished, cartoon")
+    with gr.Row():
+                    a1 = gr.Slider(label="Young", value=0, step=1, minimum=-1000000, maximum=1000000, interactive=True)
+                    a2 = gr.Slider(label="Pointy Nose", value=0, step=1, minimum=-1000000, maximum=1000000, interactive=True)
+                    a3 = gr.Slider(label="Undereye Bags", value=0, step=1, minimum=-1000000, maximum=1000000, interactive=True)
 
-            with gr.Row():
-                a1 = gr.Slider(label="Young", value=0, step=1, minimum=-1000000, maximum=1000000, interactive=True)
-                a2 = gr.Slider(label="Pointy Nose", value=0, step=1, minimum=-1000000, maximum=1000000, interactive=True)
-                a3 = gr.Slider(label="Undereye Bags", value=0, step=1, minimum=-1000000, maximum=1000000, interactive=True)
+    
+    with gr.Accordion("Advanced Options", open=False):
+        with gr.Column():     
+                seed = gr.Number(value=5, label="Seed", interactive=True)
+                cfg = gr.Slider(label="CFG", value=3.0, step=0.1, minimum=0, maximum=10, interactive=True)
+                steps = gr.Slider(label="Inference Steps",  value=50, step=1, minimum=0, maximum=100, interactive=True)
+                injection_step = gr.Slider(label="Injection Step",  value=800, step=1, minimum=0, maximum=1000, interactive=True)
 
 
-            submit = gr.Button("Submit")
+                
+    submit = gr.Button("Submit")
 
-        with gr.Column():
-            gallery2 = gr.Gallery(label="Identity from Edited Model")
+        #with gr.Column():
+            #gallery2 = gr.Gallery(label="Identity from Edited Model")
 
 
 
         
-        sample.click(fn=sample_then_run, outputs=gallery1)
+    sample.click(fn=sample_then_run, outputs=gallery1)
 
-        
-        submit.click(fn=edit_inference,
-                    inputs=[prompt, negative_prompt, cfg, steps, seed, injection_step, a1, a2, a3],
-                    outputs=gallery2)
+
+    submit.click(fn=edit_inference,
+                inputs=[prompt, negative_prompt, cfg, steps, seed, injection_step, a1, a2, a3],
+                outputs=gallery2)