add interface

App_main.py

@@ -92,7 +92,7 @@ examples = [
 
 with gr.Row():
     image_input = gr.Image(type='pil', label='Input Img')
-    image_output = gr.Image(image_mode='L', shape=(32, 32), label='Segmentation Result', tool='select')
+    image_output = gr.Image(image_mode='L', label='Segmentation Result', tool='select')
 with gr.Column():
     checkpoint = gr.inputs.Radio(['WHU', 'INRIA'], label='Checkpoint')