| import gradio as gr | |
| import image_feature as func | |
| def work11(image1, image2): | |
| return func.similarity_cpu(image1, image2) | |
| # return func.infer1(image1, image2) | |
| # return func.infer4(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="pil", label="图片1"), gr.Image(type="pil", label="图片2")], | |
| outputs=[gr.Textbox(lines=3, label="推理结果")]) | |
| # | |
| # # demo = gr.Interface(fn=work, inputs="image,text", outputs="text") | |
| # | |
| demo.launch() |