app.py

# -*- coding: utf-8 -*-
"""
# File       : employ_monitor.py
# Time       ：2023/4/18 16:23
# Author     ：小吕同学
"""
import gradio as gr
from text2vec import Similarity

# 中文句向量模型(CoSENT)
sim_model = Similarity(model_name_or_path='shibing624/text2vec-base-chinese',
                       similarity_type='cosine', embedding_type='sbert')
# sim_model.get_scores()


def ai_text(sentence1, sentence2):
    score = sim_model.get_score(sentence1, sentence2)
    print("{} \t\t {} \t\t Score: {:.4f}".format(sentence1, sentence2, score))

    return score


if __name__ == '__main__':
    examples = [
        ['如何更换花呗绑定银行卡', '花呗更改绑定银行卡'],
        ['我在北京打篮球', '我是北京人，我喜欢篮球'],
        ['一个女人在看书。', '一个女人在揉面团'],
        ['一个男人在车库里举重。', '一个人在举重。'],
    ]
    input1 = gr.inputs.Textbox(lines=2, placeholder="输入第一句话...")
    input2 = gr.inputs.Textbox(lines=2, placeholder="输入第二句话...")

    output_text = gr.outputs.Textbox()
    gr.Interface(ai_text,
                 inputs=[input1, input2],
                 outputs=[output_text],
                 theme="grass",
                 title="基于 [ shibing624/text2vec-base-chinese ] 模型的相似句子判断",
                 description="计算两个句子之间的 [cosine] 余弦相似度。",
                 article="Link to <a href='https://github.com/shibing624/text2vec' style='color:blue;' target='_blank\'>Github REPO</a>",
                 examples=examples,
                 ).launch(show_error=True)