update

app.py

@@ -1,9 +1,12 @@
 import gradio as gr
 
-def generate_text(title, context):
+def origin_preditcion(title, context):
     return f"Title:{title}\nContext:{context}\n..."
 
-def generate_mutimodal(title, context, img):
+def edit_process(title, context):
+    return f"Title:{title}\nContext:{context}\n..."
+
+def add_process(title, context, img):
     return f"Title:{title}\nContext:{context}\n...{img}"
 
 with gr.Blocks() as demo:
@@ -12,8 +15,8 @@ with gr.Blocks() as demo:
     # 多个tab
     with gr.Tabs():
 
-        with gr.TabItem("E-FB15k237", css="#warning {color: red}"):
-            title = gr.Textbox(label="Input", lines=1, placeholder="Mask triple input")
+        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="")
 
@@ -21,6 +24,13 @@ with gr.Blocks() as demo:
             edit_button = gr.Button("Edit", elem_id="warning")
 
             edit_output = gr.Textbox(label="After Edit", lines=2, placeholder="")
+            gr.Examples(
+                examples=["[MASK] r t", "h"],
+                inputs=[input, alter_label],
+                outputs=edit_output,
+                fn=edit_process,
+                cache_examples=True,
+            )
 
         with gr.TabItem("A-FB15k237"):
             title = gr.Textbox(label="Input", lines=1, placeholder="New triple input")
@@ -31,7 +41,8 @@ with gr.Blocks() as demo:
             add_output = gr.Textbox(label="Add Results", lines=2, placeholder="")
 
 
-    edit_button.click(fn=generate_text, inputs=[title, alter_label], outputs=edit_output)
-    add_button.click(fn=generate_mutimodal, inputs=[title, alter_label], outputs=add_output)
+    edit_button.click(fn=edit_process, inputs=[title, alter_label], outputs=edit_output)
+    origin_button.click(fn=origin_preditcion, inputs=[title, alter_label], outputs=edit_output)
+    add_button.click(fn=add_process, inputs=[title, alter_label], outputs=add_output)
 
 demo.launch()