File size: 2,456 Bytes
a0a66d7 90b5e3c f58c303 90b5e3c 6fc1c67 6dd01fc 90b5e3c 0f1ad79 3124969 8fa6796 d35c449 8fa6796 6dd01fc 0f1ad79 42a5ea2 d35c449 8fa6796 90b5e3c 1fb62f8 3124969 f58c303 90b5e3c 6dd01fc 3fae30f 3124969 6dd01fc 3124969 3fae30f 6dd01fc 2c36eb0 6fc1c67 6dd01fc f889cdb 3124969 6fc1c67 f889cdb 6fc1c67 3124969 6dd01fc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 |
import gradio as gr
def edit_process(title, context):
return f"Title:{title}\nContext:{context}\n...", f"Title:{title}\nContext:{context}\n..."
def add_process(title, context, img):
return f"Title:{title}\nContext:{context}\n...{img}", f"Title:{title}\nContext:{context}\n...{img}"
with gr.Blocks() as demo:
gr.Markdown("# KGE Editing")
# 多个tab
with gr.Tabs():
with gr.TabItem("E-FB15k237"):
with gr.Row():
with gr.Column():
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")
with gr.Column():
origin_output = gr.Textbox(label="Before Edit", lines=3, placeholder="")
edit_output = gr.Textbox(label="After Edit", lines=3, placeholder="")
gr.Examples(
examples=[["[MASK] r1 t1", "h1"], ["[MASK] r2 t2", "h2"]],
inputs=[edit_input, alter_label],
outputs=[origin_output, edit_output],
fn=edit_process,
cache_examples=True,
)
with gr.TabItem("A-FB15k237"):
with gr.Row():
with gr.Column():
add_input = gr.Textbox(label="Input", lines=1, placeholder="New triple input")
mask_head = gr.Textbox(label="Head/Tail", lines=1, placeholder="1:head / 0:tail")
add_button = gr.Button("Add")
with gr.Column():
add_origin_output = gr.Textbox(label="Origin Results", lines=3, placeholder="")
add_output = gr.Textbox(label="Add Results", lines=3, placeholder="")
gr.Examples(
examples=[["h1 r1 t1", "1"], ["h2 r2 t2", "1"]],
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=[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() |