Update app.py

app.py

@@ -23,8 +23,8 @@ def predict(dict, prompt="", negative_prompt="", guidance_scale=7.5, steps=30, s
     
     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 == "Realistic_V5.1":
+        pipe = AutoPipelineForInpainting.from_pretrained("SG161222/Realistic_Vision_V5.1_noVAE", vae=vae).to("cuda")
     if model == "EpicRealism":
         pipe = AutoPipelineForInpainting.from_pretrained("emilianJR/epiCRealism", vae=vae).to("cuda")
     if model == "Realistic_V6.0":
@@ -135,7 +135,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=512, width=512)
+                    image_out = gr.Image(label="Output", elem_id="output-img", height=1024, width=512)
             
 
     btn.click(fn=predict, inputs=[image, prompt, negative_prompt, guidance_scale, steps, strength, model, scheduler], outputs=[image_out], api_name='run')