Update app.py

app.py

@@ -26,12 +26,12 @@ def predict(dict, prompt="", negative_prompt="", guidance_scale=7.5, steps=30, s
 
     add_kwargs = {}
     if len(scheduler.split("-")) > 1:
-        add_kwargs["use_karras"] = True
+        add_kwargs["use_karras_sigmas"] = True
     if len(scheduler.split("-")) > 2:
         add_kwargs["algorithm_type"] = "sde-dpmsolver++"
 
     scheduler = getattr(diffusers, scheduler_class_name)
-    pipe.scheduler = scheduler.from_pretrained("runwayml/stable-diffusion-inpainting", subfolder="scheduler", **add_kwargs)
+    pipe.scheduler = scheduler.from_pretrained("emilianJR/epiCRealism", subfolder="scheduler", **add_kwargs)
 
     
     
@@ -51,7 +51,16 @@ def predict(dict, prompt="", negative_prompt="", guidance_scale=7.5, steps=30, s
     mask_image.putdata(new_data)
     mask_image = mask_image.resize(init_image.size, resample=Image.LANCZOS)
     #mask_image = pipe.mask_processor.blur(mask_image, blur_factor=25)
-    output = pipe(prompt = prompt, negative_prompt=negative_prompt, image=init_image, mask_image=mask_image, guidance_scale=guidance_scale, num_inference_steps=int(steps), strength=strength, clip_skip=1)
+    output = pipe(
+        prompt = prompt,
+        negative_prompt=negative_prompt,
+        image=init_image,
+        mask_image=mask_image,
+        guidance_scale=guidance_scale,
+        num_inference_steps=int(steps),
+        strength=strength,
+        clip_skip=1
+    )
     print("Positive:",prompt)
     print("Negative:",negative_prompt)
     print("Guidance_scale:",guidance_scale)
@@ -103,7 +112,7 @@ with image_blocks as demo:
     gr.HTML(read_content("header.html"))
     with gr.Row():
                 with gr.Column():
-                    image = gr.Image(source='upload', tool='sketch', elem_id="image_upload", type="pil", label="Upload",height=400)
+                    image = gr.Image(source='upload', tool='sketch', elem_id="image_upload", type="pil", label="Upload",height=512)
                     with gr.Row(elem_id="prompt-container",equal_height=True):
                         with gr.Row():
                             prompt = gr.Textbox(placeholder="Your prompt", show_label=False, elem_id="prompt", lines=5, allow_resize=True)
@@ -112,7 +121,7 @@ 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=100, step=1, label="steps")
+                            steps = gr.Number(value=40, minimum=10, maximum=100, step=1, label="steps")
                             strength = gr.Number(value=0.8, 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):
@@ -120,7 +129,7 @@ with image_blocks as demo:
                             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)
+                    image_out = gr.Image(label="Output", elem_id="output-img", height=512, width=512)
                     with gr.Group(elem_id="share-btn-container", visible=False) as share_btn_container:
                         community_icon = gr.HTML(community_icon_html)
                         loading_icon = gr.HTML(loading_icon_html)
@@ -132,4 +141,4 @@ with image_blocks as demo:
     share_button.click(None, [], [], _js=share_js)
 
 
-image_blocks.queue(max_size=25,api_open=True).launch(show_api=True)
+image_blocks.queue(max_size=25,api_open=True).launch(show_api=True,share=True)