change layout

app.py

@@ -162,10 +162,10 @@ def main():
                         generate_button = gr.Button("Generate")
 
             with gr.Column():
+                richtext_result = gr.Image(label='Rich-text')
                 with gr.Row():
                     plaintext_result = gr.Image(label='Plain-text')
-                    richtext_result = gr.Image(label='Rich-text')
-                token_map = gr.Image(label='Token Maps')
+                    token_map = gr.Image(label='Token Maps')
 
         with gr.Row():
             gr.Markdown(help_text)