Update app.py

app.py

@@ -433,70 +433,109 @@ intro = """
 
 with gr.Blocks(css="style.css") as demo:
     gr.HTML(intro)
-    
-    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`
-    """)
-    with gr.Column():
-            with gr.Row():
-                with gr.Column():
-                    sample = gr.Button("🎲 Sample New Model")
-                    file_input = gr.File(label="Upload Model", container=True)
-                    file_output = gr.File(label="Download Sampled Model", container=True, interactive=False)
+    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`
+        """)
+        with gr.Column():
+                with gr.Row():
+                    with gr.Column():
+                        sample = gr.Button("🎲 Sample New Model")
+                        file_output = gr.File(label="Download Sampled Model", container=True, interactive=False)
+                        file_input = gr.File(label="Upload Model", container=True)
+
+
+                            # invert_button = gr.Button("⏪ Invert")
+                    with gr.Column():
+                        image_slider = ImageSlider(position=0.5, type="pil", height=512, width=512)
+                        
+                        prompt1 = gr.Textbox(label="Prompt",
+                                                        info="Make sure to include 'sks person'" ,
+                                                        placeholder="sks person", 
+                                                        value="sks person")
+                        seed1 = gr.Number(value=5, label="Seed", precision=0, interactive=True)
 
+                    
+                
 
-                        # invert_button = gr.Button("⏪ Invert")
-                with gr.Column():
-                    image_slider = ImageSlider(position=0.5, type="pil", height=512, width=512)
+                        with gr.Row():
+                            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="- Thick Eyebrows +", value=0, step=0.001, minimum=-1, maximum=1, interactive=True)
+
+                
+                
+                        with gr.Accordion("Advanced Options", open=False):
+                            cfg1= gr.Slider(label="CFG", value=3.0, step=0.1, minimum=0, maximum=10, interactive=True)
+                            steps1 = gr.Slider(label="Inference Steps",  value=50, step=1, minimum=0, maximum=100, interactive=True)
+                            negative_prompt1 = gr.Textbox(label="Negative Prompt", placeholder="low quality, blurry, unfinished, nudity, weapon", value="low quality, blurry, unfinished, nudity, weapon")
+                            injection_step = gr.Slider(label="Injection Step",  value=800, step=1, minimum=0, maximum=1000, interactive=True)
+                                
+                                
                     
-                    prompt1 = gr.Textbox(label="Prompt",
-                                                    info="Make sure to include 'sks person'" ,
-                                                    placeholder="sks person", 
-                                                    value="sks person")
-                    seed1 = gr.Number(value=5, label="Seed", precision=0, interactive=True)
 
+                        submit1 = gr.Button("Generate")
+
+    with gr.Tab("Inversion"):
+        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) 
+            lr = gr.Number(value=1e-1, label="Learning Rate", interactive=True)
+            pcs = gr.Slider(label="# Principal Components", value=10000, step=1, minimum=1, maximum=10000, interactive=True)
+            epochs = gr.Slider(label="Epochs", value=400, step=1, minimum=1, maximum=2000, interactive=True)
+            weight_decay = gr.Number(value=1e-10, label="Weight Decay", interactive=True)
+
+        with gr.Column():
+            image_slider = ImageSlider(position=0.5, type="pil", height=512, width=512)
+                        
+            prompt1 = gr.Textbox(label="Prompt",
+                                info="Make sure to include 'sks person'" ,
+                                placeholder="sks person", 
+                                value="sks person")
+            seed1 = gr.Number(value=5, label="Seed", precision=0, interactive=True)
+
+                    
                 
-            
 
-                    with gr.Row():
-                        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="- Thick Eyebrows +", value=0, step=0.001, minimum=-1, maximum=1, interactive=True)
+            with gr.Row():
+                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="- Thick Eyebrows +", value=0, step=0.001, minimum=-1, maximum=1, interactive=True)
 
-            
-            
-                    with gr.Accordion("Advanced Options", open=False):
-                        cfg1= gr.Slider(label="CFG", value=3.0, step=0.1, minimum=0, maximum=10, interactive=True)
-                        steps1 = gr.Slider(label="Inference Steps",  value=50, step=1, minimum=0, maximum=100, interactive=True)
-                        negative_prompt1 = gr.Textbox(label="Negative Prompt", placeholder="low quality, blurry, unfinished, nudity, weapon", value="low quality, blurry, unfinished, nudity, weapon")
-                        injection_step = gr.Slider(label="Injection Step",  value=800, step=1, minimum=0, maximum=1000, interactive=True)
-                            
-                            
                 
+                
+            with gr.Accordion("Advanced Options", open=False):
+                cfg1= gr.Slider(label="CFG", value=3.0, step=0.1, minimum=0, maximum=10, interactive=True)
+                steps1 = gr.Slider(label="Inference Steps",  value=50, step=1, minimum=0, maximum=100, interactive=True)
+                negative_prompt1 = gr.Textbox(label="Negative Prompt", placeholder="low quality, blurry, unfinished, nudity, weapon", value="low quality, blurry, unfinished, nudity, weapon")
+                injection_step = gr.Slider(label="Injection Step",  value=800, step=1, minimum=0, maximum=1000, interactive=True)
+                                
+                                
+                    
 
-                    submit1 = gr.Button("Generate")
-     
+            submit2 = gr.Button("Generate")
 
+        
+                    
+        
 
-   
-            
+
+    
+                
     sample.click(fn=sample_then_run, outputs=[image_slider, file_output])
 
-       
+        
     submit1.click(fn=edit_inference, inputs=[ prompt1, negative_prompt1, cfg1, steps1, seed1, injection_step, a1, a2, a3, a4], outputs=image_slider)
     file_input.change(fn=file_upload, inputs=file_input, outputs = image_slider)
 
 
 
 
-
-    
-    
-