diff --git a/draft.py b/draft.py
index 95b7aef6..c507b2e3 100644
--- a/draft.py
+++ b/draft.py
@@ -20,7 +20,8 @@ with block:
     with gr.Column():
         prompt = gr.Textbox(label="Prompt", value='a handsome man in forest')
         run_button = gr.Button(label="Run")
-        result_gallery = gr.Gallery(label='Output', show_label=False, elem_id="gallery", height='auto')
+        result_gallery = gr.Gallery(label='Output', show_label=False, elem_id="gallery",
+                                    object_fit='contain', height=768)
     run_button.click(fn=generate_clicked, inputs=[prompt], outputs=[result_gallery])
 
 
diff --git a/modules/default_pipeline.py b/modules/default_pipeline.py
index 9634f994..7db45848 100644
--- a/modules/default_pipeline.py
+++ b/modules/default_pipeline.py
@@ -18,7 +18,7 @@ xl_refiner = core.load_model(xl_refiner_filename)
 
 
 @torch.no_grad()
-def process(positive_prompt, negative_prompt, width=1024, height=1024, batch_size=1):
+def process(positive_prompt, negative_prompt, width=1280, height=960, batch_size=1):
     positive_conditions = core.encode_prompt_condition(clip=xl_base.clip, prompt=positive_prompt)
     negative_conditions = core.encode_prompt_condition(clip=xl_base.clip, prompt=negative_prompt)