Update app.py

app.py

@@ -9,7 +9,7 @@ import diffusers
 
 
 vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse")
-pipe = AutoPipelineForInpainting.from_pretrained("SG161222/Realistic_Vision_V5.0_noVAE",vae=vae).to("cuda")
+
 
 def read_content(file_path: str) -> str:
     """read the content of target file
@@ -19,7 +19,18 @@ def read_content(file_path: str) -> str:
 
     return content
 
-def predict(dict, prompt="", negative_prompt="", guidance_scale=7.5, steps=30, strength=0.8, scheduler="DPMSolverMultistepScheduler-Karras"):
+def predict(dict, prompt="", negative_prompt="", guidance_scale=7.5, steps=30, strength=0.8,model="Realistic_V5.0", scheduler="DPMSolverMultistepScheduler-Karras"):
+    
+    pipe = AutoPipelineForInpainting.from_pretrained("SG161222/Realistic_Vision_V5.0_noVAE",vae=vae).to("cuda")
+    
+    if model == "Realistic_V5.0":
+        pipe = AutoPipelineForInpainting.from_pretrained("SG161222/Realistic_Vision_V5.0_noVAE", vae=vae).to("cuda")
+    if model == "EpicRealism":
+        pipe = AutoPipelineForInpainting.from_pretrained("emilianJR/epiCRealism", vae=vae).to("cuda")
+    if model == "Realistic_V6.0":
+        pipe = AutoPipelineForInpainting.from_pretrained("SG161222/Realistic_Vision_V6.0_B1_noVAE", vae=vae).to("cuda")
+
+        
     if negative_prompt == "":
         negative_prompt = None
     scheduler_class_name = scheduler.split("-")[0]
@@ -50,7 +61,7 @@ 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)
+    mask_image = pipe.mask_processor.blur(mask_image, blur_factor=5)
     output = pipe(
         prompt = prompt,
         negative_prompt=negative_prompt,
@@ -105,9 +116,9 @@ with image_blocks as demo:
                     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)
-                        with gr.Row():
-                            btn = gr.Button("Inpaint!", elem_id="run_button")
+                            prompt = gr.Textbox(placeholder="Your prompt", show_label=False, elem_id="prompt", lines=5)
+                    with gr.Row(equal_height=True):
+                        btn = gr.Button("Inpaint!", elem_id="run_button")
                     
                     with gr.Accordion(label="Advanced Settings", open=False):
                         with gr.Row(equal_height=True):