feat(app.py): update app

.gitignore

@@ -0,0 +1,2 @@
+**/__pycache__
+flagged

app.py

@@ -0,0 +1,26 @@
+from transformers.pipelines import pipeline
+import gradio as gr
+from huggingface_hub import login
+import os
+login(os.environ['HF_Token'])
+predictor = pipeline(
+    "sentences_sim",
+    model="minskiter/simbert-chinese-bert-wwm-ext",
+    device="cpu",
+    trust_remote_code=True,
+    use_auth_token=True
+)
+
+def sim_predictor_gradio(text_a,text_b):
+    sim = predictor((text_a, text_b))
+    return {text_b[:10]+"...": sim}
+
+demo = gr.Interface(
+    fn=sim_predictor_gradio, 
+    inputs=[
+        gr.Textbox(lines=5, label="文本1"),
+        gr.Textbox(lines=5, label="文本2")
+    ], 
+    outputs=gr.Label(num_top_classes=1, label="相似度")
+)
+demo.launch()

requirements.txt

@@ -0,0 +1,4 @@
+transformers==4.30.1
+gradio==3.36.1
+huggingface-hub==0.15.1
+torch==2.0.1