Update app.py

app.py

@@ -450,91 +450,44 @@ with gr.Blocks(css="style.css") as demo:
                         # invert_button = gr.Button("⏪ Invert")
                 with gr.Column():
                     image_slider = ImageSlider(position=0.5, type="pil", height=512, width=512)
-                    # gallery1 = gr.Image(label="Identity from Original Model",height=512, width=512, interactive=False)
                     
-            prompt1 = gr.Textbox(label="Prompt",
-                                            info="Make sure to include 'sks person'" ,
-                                            placeholder="sks person", 
-                                            value="sks person")
-                
-            # Editing 
-            with gr.Column():
-                #gallery2 = gr.Image(label="Identity from Edited Model", interactive=False, visible=False )
-                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)
-#                 prompt2 = gr.Textbox(label="Prompt",
-#                                         info="Make sure to include 'sks person'" ,
-#                                         placeholder="sks person", 
-#                                         value="sks person", visible=False)
-#                 seed2 = gr.Number(value=5, label="Seed", precision=0, interactive=True, visible=False)
-
+                    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)
 
+                
+            
 
-#                 submit2 = gr.Button("Generate", visible=False)
+                    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):
-                with gr.Tab("Inversion"):
-                    with gr.Row():     
-                        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)
-                    with gr.Row():
-                        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.Tab("Sampling"):
-                    with gr.Row():     
-                            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)
-                            seed1 = gr.Number(value=5, label="Seed", precision=0, interactive=True)
-                    with gr.Row():
-                            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)
                             
                             
-                # with gr.Tab("Editing"):
-                #     with gr.Column():     
-                        # cfg2 = gr.Slider(label="CFG", value=3.0, step=0.1, minimum=0, maximum=10, interactive=True)
-                        # steps2 = 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)
-                        # negative_prompt2 = gr.Textbox(label="Negative Prompt", placeholder="low quality, blurry, unfinished, nudity, weapon", value="low quality, blurry, unfinished, nudity, weapon")
-                with gr.Tab("Uploading a model"):
-                    gr.Markdown("""<div style="text-align: justify;">Upload a model below downloaded from this demo.""")
-
-                    file_input = gr.File(label="Upload Model", container=True)
+                
 
             submit1 = gr.Button("Generate")
      
 
 
-    # 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)
-        
-
-
-
-
-        # invert_button.click(fn=run_inversion,
-        #             inputs=[input_image, pcs, epochs, weight_decay,lr], 
-        #             outputs = [image_slider, file_output])
-        
+   
             
     sample.click(fn=sample_then_run, outputs=[image_slider, file_output])
 
-        # submit1.click(fn=inference,
-        #             inputs=[prompt1, negative_prompt1, cfg1, steps1, seed1],
-        #             outputs=gallery1)
-        # submit1.click(fn=edit_inference,
-        #             inputs=[input_image, prompt1, negative_prompt1, cfg1, steps1, seed1, injection_step, a1, a2, a3, a4],
-        #             outputs=image_slider)
-    submit1.click(
-    fn=edit_inference, inputs=[ prompt1, negative_prompt1, cfg1, steps1, seed1, injection_step, a1, a2, a3, a4], outputs=image_slider)
+       
+    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)