update

app.py

@@ -1,8 +1,10 @@
 import paddlehub as hub
 import gradio as gr
 
+
 module = hub.Module(name="lseg")
 
+
 def segment(image, labels):
     results = module.segment(
         image=image[..., ::-1],
@@ -10,4 +12,10 @@ def segment(image, labels):
     )
     return [results['color'][..., ::-1], results['mix'][..., ::-1], *[v[..., ::-1] for v in results['classes'].values()]]
 
-gr.Interface(fn=segment, inputs=['image', gr.Textbox(placeholder='other\ncat', lines=5, max_lines=50)], outputs=['gallery']).launch()
+
+gr.Interface(
+    fn=segment,
+    inputs=['image', gr.Textbox(
+        placeholder='other\ncat', lines=5, max_lines=50)],
+    outputs=[gr.Gallery().style(grid=[2, 3], height="auto")]
+).launch()