test `output_gallery_component` as output

app.py

@@ -98,7 +98,7 @@ with gr.Blocks() as demo:
                     value='Submit', variant='primary', scale=0)
         with gr.Column():
             output_gallery_component = gr.Gallery(
-                label='Generated image')
+                object_fit="contain", height="auto", label='Generated image')
 
     submit_button_component.click(
         fn=process,