Update app.py

app.py

@@ -29,10 +29,10 @@ def predict(dict, prompt="", negative_prompt="", guidance_scale=7.5, steps=20, s
     scheduler = getattr(diffusers, scheduler_class_name)
     pipe.scheduler = scheduler.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", subfolder="scheduler", **add_kwargs)
     
-    init_image = dict["image"].convert("RGB")
-    mask = dict["mask"].convert("RGB")
+    init_image = dict["image"]
+    mask = dict["mask"]
     
-    output = pipe(prompt = prompt, negative_prompt=negative_prompt, image=init_image, mask_image=mask, guidance_scale=guidance_scale, num_inference_steps=int(steps), strength=strength)
+    output = pipe(prompt = prompt, negative_prompt=negative_prompt, image=init_image, mask_image=mask, guidance_scale=guidance_scale, num_inference_steps=int(steps), strength=strength, clip_skip=1)
     
     return output.images[0], gr.update(visible=True)
 
@@ -88,12 +88,12 @@ with image_blocks as demo:
                     with gr.Accordion(label="Advanced Settings", open=False):
                         with gr.Row(equal_height=True):
                             guidance_scale = gr.Number(value=7.5, minimum=1.0, maximum=20.0, step=0.1, label="guidance_scale")
-                            steps = gr.Number(value=20, minimum=10, maximum=30, step=1, label="steps")
+                            steps = gr.Number(value=20, minimum=10, maximum=100, step=1, label="steps")
                             strength = gr.Number(value=0.99, minimum=0.01, maximum=1.0, step=0.01, label="strength")
                             negative_prompt = gr.Textbox(label="negative_prompt", placeholder="Your negative prompt", info="what you don't want to see in the image")
                         with gr.Row(equal_height=True):
                             schedulers = ["DEISMultistepScheduler", "HeunDiscreteScheduler", "EulerDiscreteScheduler", "DPMSolverMultistepScheduler", "DPMSolverMultistepScheduler-Karras", "DPMSolverMultistepScheduler-Karras-SDE"]
-                            scheduler = gr.Dropdown(label="Schedulers", choices=schedulers, value="EulerDiscreteScheduler")
+                            scheduler = gr.Dropdown(label="Schedulers", choices=schedulers, value="DPMSolverMultistepScheduler-Karras")
                         
                 with gr.Column():
                     image_out = gr.Image(label="Output", elem_id="output-img", height=400)