Update app.py

app.py

@@ -27,37 +27,23 @@ def divisible_by_8(image):
     return resized_image
 
 
-def generate(image_editor, prompt, neg_prompt, strength, guidance):
+def generate(image_editor, prompt, neg_prompt):
     image = image_editor['background'].convert('RGB')
     image.thumbnail((1024, 1024))
     image = divisible_by_8(image)
-
     original_image_size = image.size
     
-
     layer = image_editor["layers"][0].resize(image.size)
-
     
     image = squarify_image(image)
-    #layer = squarify_image(layer)
     
     mask = Image.new("RGBA", image.size, "WHITE") 
     mask.paste(layer, (0, 0), layer)
     mask = ImageOps.invert(mask.convert('L'))
 
-
-    
-    
-
     final_image = pipeline(prompt=prompt, 
                            image=image, 
                            mask_image=mask).images[0]
-                           #width=image.width, 
-                           #height=image.height, 
-                           #num_inference_steps=50, 
-                           #strength=strength, 
-                           #guidance_scale=guidance).images[0]
-
     
     final_image = final_image.crop((0, 0, original_image_size[0], original_image_size[1]))
     
@@ -77,17 +63,14 @@ with gr.Blocks() as demo:
         with gr.Column():
             version_gallery = gr.Gallery(label="Versions")
             restore_button = gr.Button("Restore Version")
-
-    with gr.Accordion("Advanced Settings", open=False):
-        neg_prompt = gr.Textbox(label='Negative Prompt', value='ugly, deformed, nsfw')
-        strength_slider = gr.Slider(0.0, 1.0, value=1.0, label="Strength")
-        guidance_slider = gr.Slider(1.0, 15.0, value=7.5, label="Guidance")
+            with gr.Accordion("Advanced Settings", open=False):
+                neg_prompt = gr.Textbox(label='Negative Prompt', value='ugly, deformed, nsfw')
         
     with gr.Row():
         out1 = gr.Image(format='png')
         out2 = gr.Image(format='png')
         out3 = gr.Image(format='png')
         
-    generate_button.click(fn=generate, inputs=[sketch_pad,prompt, neg_prompt, strength_slider, guidance_slider], outputs=[sketch_pad, out1, out2, out3])
+    generate_button.click(fn=generate, inputs=[sketch_pad,prompt, neg_prompt], outputs=[sketch_pad, out1, out2, out3])
 
 demo.launch()