initial commit

.gitignore

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

app.py

@@ -0,0 +1,8 @@
+import gradio as gr
+import model
+
+def encode(text):
+    return model.encode(text)
+
+iface = gr.Interface(fn=encode, inputs="text", outputs="text")
+iface.launch()

model.py

@@ -0,0 +1,7 @@
+from sentence_transformers import SentenceTransformer
+model = SentenceTransformer('moka-ai/m3e-base')
+# model = SentenceTransformer('nghuyong/ernie-3.0-base-zh')
+
+def encode(text: str):
+    text = text.replace("\n", " ")
+    return model.encode([text], normalize_embeddings=True).tolist()[0]

requirements.txt

@@ -0,0 +1 @@
+sentence-transformers==2.2.2