Update app.py

app.py

@@ -46,14 +46,14 @@ def inference(prompt, guidance, steps, image_size="Square", seed=0, img=None, st
   prompt,neg_prompt=auto_prompt_correction(prompt,neg_prompt,disable_auto_prompt_correction)
 
   if(image_size=="Portrait"):
-      height=1024
+      height=1344
       width=768
   elif(image_size=="Landscape"):
       height=768
-      width=1024
+      width=1344
   else:
-      height=768
-      width=768
+      height=1024
+      width=1024
 
   print(prompt,neg_prompt)
     
@@ -167,7 +167,7 @@ with gr.Blocks(css=css) as demo:
                 prompt = gr.Textbox(label="Prompt", show_label=False, max_lines=2,placeholder="[your prompt]").style(container=False)
                 generate = gr.Button(value="Generate")
 
-              image_out = gr.Image(height=768,width=768)
+              image_out = gr.Image(height=1024,width=1024)
           error_output = gr.Markdown()
 
         with gr.Column(scale=45):