embeddings-api-ernie

Sleeping

justest commited on Sep 10, 2023

Commit

6e6e517

1 Parent(s): b19a023

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -1,6 +1,6 @@
 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", " ")

 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", " ")