Update app.py

app.py

@@ -99,8 +99,9 @@ def inference( prompt, negative_prompt, guidance_scale, ddim_steps, seed):
     return [image] 
 
 
+
 @torch.no_grad()
-def edit_inference(prompt, negative_prompt, guidance_scale, ddim_steps, seed, start_noise, a1, a2, a3):
+def edit_inference(prompt, negative_prompt, guidance_scale, ddim_steps, seed, start_noise, a1, a2, a3, a4):
 
     global device
     global generator 
@@ -111,13 +112,14 @@ def edit_inference(prompt, negative_prompt, guidance_scale, ddim_steps, seed, st
     global noise_scheduler
     global young
     global pointy
-    global bags
+    global wavy
+    global large
     
     original_weights = network.proj.clone()
     
 
-    edited_weights = original_weights+a1*young+a2*pointy+a3*bags
-
+    edited_weights = original_weights+a1*1e6*young+a2*1e6*pointy+a3*1e6*wavy+a4*2e6*large
+   
     generator = generator.manual_seed(seed)
     latents = torch.randn(
         (1, unet.in_channels, 512 // 8, 512 // 8),
@@ -204,20 +206,43 @@ def sample_then_run():
 #directions
 global young
 global pointy
-global bags
+global wavy 
+global large
+
 young = get_direction(df, "Young", pinverse, 1000, device)
 young = debias(young, "Male", df, pinverse, device)
+young = debias(young, "Pointy_Nose", df, pinverse, device)
+young = debias(young, "Wavy_Hair", df, pinverse, device)
+young = debias(young, "Chubby", df, pinverse, device)
 young_max = torch.max(proj@young[0]/(torch.norm(young))**2).item()
 young_min = torch.min(proj@young[0]/(torch.norm(young))**2).item()
 
 pointy = get_direction(df, "Pointy_Nose", pinverse, 1000, device)
+pointy = debias(pointy, "Young", df, pinverse, device)
+pointy = debias(pointy, "Male", df, pinverse, device)
+pointy = debias(pointy, "Wavy_Hair", df, pinverse, device)
+pointy = debias(pointy, "Chubby", df, pinverse, device)
+pointy = debias(pointy, "Heavy_Makeup", df, pinverse, device)
 pointy_max = torch.max(proj@pointy[0]/(torch.norm(pointy))**2).item()
 pointy_min = torch.min(proj@pointy[0]/(torch.norm(pointy))**2).item()
 
-bags = get_direction(df, "Bags_Under_Eyes", pinverse, 1000, device)
-bags_max = torch.max(proj@bags[0]/(torch.norm(bags))**2).item()
-bags_min = torch.min(proj@bags[0]/(torch.norm(bags))**2).item()
 
+wavy = get_direction(df, "Wavy_Hair", pinverse, 1000, device)
+wavy = debias(wavy, "Young", df, pinverse, device)
+wavy = debias(wavy, "Male", df, pinverse, device)
+wavy = debias(wavy, "Pointy_Nose", df, pinverse, device)
+wavy = debias(wavy, "Chubby", df, pinverse, device)
+wavy = debias(wavy, "Heavy_Makeup", df, pinverse, device)
+wavy_max = torch.max(proj@wavy[0]/(torch.norm(wavy))**2).item()
+wavy_min = torch.min(proj@wavy[0]/(torch.norm(wavy))**2).item()
+
+large = get_direction(df, "Chubby", pinverse, 1000, device)
+large = debias(large, "Male", df, pinverse, device)
+large = debias(large, "Young", df, pinverse, device)
+large = debias(large, "Pointy_Nose", df, pinverse, device)
+large = debias(large, "Wavy_Hair", df, pinverse, device)
+large_max = torch.max(proj@large[0]/(torch.norm(large))**2).item()
+large_min = torch.min(proj@large[0]/(torch.norm(large))**2).item()
 
 
 intro = """
@@ -252,9 +277,11 @@ with gr.Blocks(css="style.css") as demo:
                             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)
+                    a1 = gr.Slider(label="+Young", value=0, step=0.001, minimum=-1, maximum=1, interactive=True)
+                    a2 = gr.Slider(label="+Pointy Nose", value=0, step=0.001, minimum=-1, maximum=1, interactive=True)
+    with gr.Row():                
+                    a3 = gr.Slider(label="+Curly Hair", value=0, step=0.001, minimum=-1, maximum=1, interactive=True)
+                    a4 = gr.Slider(label="+Large", value=0, step=0.001, minimum=-1, maximum=1, interactive=True)
 
     
     with gr.Accordion("Advanced Options", open=False):
@@ -268,8 +295,8 @@ with gr.Blocks(css="style.css") as demo:
                 
     submit = gr.Button("Submit")
 
-        #with gr.Column():
-            #gallery2 = gr.Gallery(label="Identity from Edited Model")
+
+
 
 
 
@@ -278,7 +305,7 @@ with gr.Blocks(css="style.css") as demo:
 
 
     submit.click(fn=edit_inference,
-                inputs=[prompt, negative_prompt, cfg, steps, seed, injection_step, a1, a2, a3],
+                inputs=[prompt, negative_prompt, cfg, steps, seed, injection_step, a1, a2, a3, a4],
                 outputs=gallery2)
             
 
@@ -289,6 +316,5 @@ with gr.Blocks(css="style.css") as demo:
 demo.launch(share=True)
 
 
-