|
import gradio as gr |
|
|
|
def generate_text(title, context): |
|
return f"Title:{title}\nContext:{context}\n..." |
|
|
|
def generate_mutimodal(title, context, img): |
|
return f"Title:{title}\nContext:{context}\n...{img}" |
|
|
|
with gr.Blocks() as demo: |
|
gr.Markdown("# KGE Editing") |
|
|
|
|
|
with gr.Tabs(): |
|
|
|
with gr.TabItem("E-FB15k237"): |
|
title = 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="") |
|
|
|
alter_label = gr.Textbox(label="Alter Entity", lines=1, placeholder="Entity Name") |
|
edit_button = gr.Button("Edit") |
|
|
|
edit_output = gr.Textbox(label="After Edit", lines=2, placeholder="") |
|
|
|
with gr.TabItem("A-FB15k237"): |
|
title = 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") |
|
|
|
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) |
|
|
|
demo.launch() |
