app.py

app.py

@@ -53,7 +53,7 @@ def make_me():
             stop_button.click(on_stop_click, inputs=None, outputs=[gen_button, stop_button])
     
     with gr.Row():
-        output = [gr.Image(label=m, width=512, height=None, elem_id="custom_image", show_label=True, interactive=False, show_share_button=False) for m in default_models]
+        output = [gr.Image(label=m, width=None, height=None, elem_id="custom_image", show_label=True, interactive=False, show_share_button=False) for m in default_models]
         current_models = [gr.Textbox(m, visible=False) for m in default_models]
         for m, o in zip(current_models, output):
             gen_event = gen_button.click(gen_fn, [m, txt_input], o)