diff --git a/draft.py b/draft.py
index d987873..95b7aef 100644
--- a/draft.py
+++ b/draft.py
@@ -20,7 +20,7 @@ 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=1280)
+        result_gallery = gr.Gallery(label='Output', show_label=False, elem_id="gallery", height='auto')
     run_button.click(fn=generate_clicked, inputs=[prompt], outputs=[result_gallery])