Add new SentenceTransformer model.
Browse files
README.md
CHANGED
|
@@ -34,7 +34,10 @@ Then you can use the model using the original model name like this:
|
|
| 34 |
|
| 35 |
```python
|
| 36 |
from light_embed import TextEmbedding
|
| 37 |
-
sentences = [
|
|
|
|
|
|
|
|
|
|
| 38 |
|
| 39 |
model = TextEmbedding('sentence-transformers/all-MiniLM-L6-v2')
|
| 40 |
embeddings = model.encode(sentences)
|
|
@@ -45,7 +48,10 @@ Then you can use the model using onnx model name like this:
|
|
| 45 |
|
| 46 |
```python
|
| 47 |
from light_embed import TextEmbedding
|
| 48 |
-
sentences = [
|
|
|
|
|
|
|
|
|
|
| 49 |
|
| 50 |
model = TextEmbedding('LightEmbed/sbert-all-MiniLM-L6-v2-onnx')
|
| 51 |
embeddings = model.encode(sentences)
|
|
|
|
| 34 |
|
| 35 |
```python
|
| 36 |
from light_embed import TextEmbedding
|
| 37 |
+
sentences = [
|
| 38 |
+
"This is an example sentence",
|
| 39 |
+
"Each sentence is converted"
|
| 40 |
+
]
|
| 41 |
|
| 42 |
model = TextEmbedding('sentence-transformers/all-MiniLM-L6-v2')
|
| 43 |
embeddings = model.encode(sentences)
|
|
|
|
| 48 |
|
| 49 |
```python
|
| 50 |
from light_embed import TextEmbedding
|
| 51 |
+
sentences = [
|
| 52 |
+
"This is an example sentence",
|
| 53 |
+
"Each sentence is converted"
|
| 54 |
+
]
|
| 55 |
|
| 56 |
model = TextEmbedding('LightEmbed/sbert-all-MiniLM-L6-v2-onnx')
|
| 57 |
embeddings = model.encode(sentences)
|