File size: 2,435 Bytes
a0a66d7 90b5e3c f58c303 90b5e3c 6fc1c67 6dd01fc 90b5e3c 0f1ad79 8fa6796 d35c449 8fa6796 6dd01fc 0f1ad79 42a5ea2 d35c449 8fa6796 90b5e3c 1fb62f8 90b5e3c f58c303 90b5e3c 6dd01fc 3fae30f 6dd01fc 3fae30f 6dd01fc 2c36eb0 6fc1c67 6dd01fc f889cdb 6fc1c67 f889cdb 6fc1c67 f58c303 6fc1c67 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():
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=[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():
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")
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=[input, alter_label],
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])
demo.launch() |