import gradio as gr | |
import image_feature as func | |
def work11(image1, image2): | |
# return func.infer1(image1, image2) | |
return func.infer3(image1, image2) | |
# with gr.Blocks() as demo: | |
# image_url = gr.Image(type="filepath", label="请选择一张图片") | |
# input = gr.Textbox(label='请描述您的问题', placeholder="", lines=1) | |
# output = gr.Textbox(label='答案', placeholder="", lines=2, interactive=False) | |
# submit = gr.Button('提问', variant="primary") | |
# submit.click(work, inputs=[image_url, input], outputs=output) | |
# demo.launch() | |
# 定义你的界面 | |
# with gr.Interface(fn=work11, | |
# inputs=[gr.Textbox(label='图片1', lines=1), gr.Textbox(label='图片2', lines=1)], # 两个文本输入框 | |
# outputs=[gr.Textbox(lines=3, label="推理结果")], # 输出为文本 | |
# title="图片相似度推理", # 界面标题 | |
# description="输入两张图片链接进行相似度推理", # 界面描述 | |
# examples=[["https://example.com", "https://google.com"], # 示例输入 | |
# ["https://github.com", "https://twitter.com"]]) as demo: # 更多示例输入 | |
# demo.launch() # 启动界面 | |
demo = gr.Interface(title="图片相似度推理", | |
css="", | |
fn=work11, | |
inputs=[gr.Image(type="filepath", label="图片1"), gr.Image(type="filepath", label="图片2")], | |
outputs=[gr.Textbox(lines=3, label="推理结果")]) | |
# | |
# # demo = gr.Interface(fn=work, inputs="image,text", outputs="text") | |
# | |
demo.launch() |