update

app.py

@@ -15,7 +15,7 @@ with gr.Blocks() as demo:
         with gr.TabItem("E-FB15k237"):
             with gr.Row():
                 with gr.Column():
-                    input = gr.Textbox(label="Input", lines=1, placeholder="Mask triple input")
+                    edit_input = gr.Textbox(label="Input", lines=1, placeholder="Mask triple input")
 
                     alter_label = gr.Textbox(label="Alter Entity", lines=1, placeholder="Entity Name")    
                     edit_button = gr.Button("Edit")
@@ -26,7 +26,7 @@ with gr.Blocks() as demo:
        
             gr.Examples(
                 examples=[["[MASK] r1 t1", "h1"], ["[MASK] r2 t2", "h2"]],
-                inputs=[input, alter_label],
+                inputs=[edit_input, alter_label],
                 outputs=[origin_output, edit_output],
                 fn=edit_process,
                 cache_examples=True,
@@ -35,9 +35,9 @@ with gr.Blocks() as demo:
         with gr.TabItem("A-FB15k237"):
             with gr.Row():
                 with gr.Column():
-                    input = gr.Textbox(label="Input", lines=1, placeholder="New triple input")
+                    add_input = gr.Textbox(label="Input", lines=1, placeholder="New triple input")
 
-                    alter_label = gr.Textbox(label="Head/Tail", lines=1, placeholder="1:head / 0:tail")
+                    mask_head = gr.Textbox(label="Head/Tail", lines=1, placeholder="1:head / 0:tail")
                     add_button = gr.Button("Add")
 
                 with gr.Column():
@@ -46,14 +46,14 @@ with gr.Blocks() as demo:
 
             gr.Examples(
                 examples=[["h1 r1 t1", "1"], ["h2 r2 t2", "1"]],
-                inputs=[input, alter_label],
+                inputs=[add_input, mask_head],
                 outputs=[add_origin_output, add_output],
                 fn=add_process,
                 cache_examples=True,
             )
 
     # origin_button.click(fn=origin_preditcion, inputs=[input, alter_label], outputs=origin_output)
-    edit_button.click(fn=edit_process, inputs=[input, alter_label], outputs=[origin_output, edit_output])
-    add_button.click(fn=add_process, inputs=[input, alter_label], outputs=[add_origin_output, add_output])
+    edit_button.click(fn=edit_process, inputs=[edit_input, alter_label], outputs=[origin_output, edit_output])
+    add_button.click(fn=add_process, inputs=[add_input, mask_head], outputs=[add_origin_output, add_output])
 
 demo.launch()