update

app.py

@@ -16,14 +16,21 @@ with gr.Blocks() as demo:
     with gr.Tabs():
 
         with gr.TabItem("E-FB15k237"):
-            input = gr.Textbox(label="Input", lines=1, placeholder="Mask triple input")
-            origin_button = gr.Button("Origin")
-            origin_output = gr.Textbox(label="Before Edit", lines=2, placeholder="")
+            with gr.Row():
+                with gr.Column():
+                    input = gr.Textbox(label="Input", lines=1, placeholder="Mask triple input")
+                    origin_button = gr.Button("Origin")
 
-            alter_label = gr.Textbox(label="Alter Entity", lines=1, placeholder="Entity Name")            
-            edit_button = gr.Button("Edit", elem_id="warning")
+                with gr.Column():
+                    origin_output = gr.Textbox(label="Before Edit", lines=2, placeholder="")
 
-            edit_output = gr.Textbox(label="After Edit", lines=2, placeholder="")
+            with gr.Row():
+                with gr.Column():
+                    alter_label = gr.Textbox(label="Alter Entity", lines=1, placeholder="Entity Name")            
+                    edit_button = gr.Button("Edit", elem_id="warning")
+
+                with gr.Column():
+                    edit_output = gr.Textbox(label="After Edit", lines=2, placeholder="")
             gr.Examples(
                 examples=[["[MASK] r1 t1", "h1"], ["[MASK] r2 t2", "h2"]],
                 inputs=[input, alter_label],
@@ -40,9 +47,8 @@ with gr.Blocks() as demo:
 
             add_output = gr.Textbox(label="Add Results", lines=2, placeholder="")
 
-
+    origin_button.click(fn=origin_preditcion, inputs=[input, alter_label], outputs=origin_output)
     edit_button.click(fn=edit_process, inputs=[input, alter_label], outputs=edit_output)
-    origin_button.click(fn=origin_preditcion, inputs=[input, alter_label], outputs=edit_output)
     add_button.click(fn=add_process, inputs=[input, alter_label], outputs=add_output)
 
 demo.launch()