File size: 2,679 Bytes
a0a66d7 90b5e3c f58c303 90b5e3c 6fc1c67 6dd01fc 90b5e3c 0f1ad79 6fc1c67 d35c449 6dd01fc 0f1ad79 42a5ea2 d35c449 6dd01fc d35c449 0f1ad79 d35c449 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 60 61 62 63 64 65 66 |
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")
# 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=3, placeholder="")
edit_output = gr.Textbox(label="After Edit", lines=3, placeholder="")
# with gr.Row():
# with gr.Column():
# with gr.Column():
#
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() |