Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -7,7 +7,7 @@ from sentence_transformers.quantization import quantize_embeddings
|
|
| 7 |
import torch.nn.functional as F
|
| 8 |
|
| 9 |
# 1. Specify preffered dimensions
|
| 10 |
-
dimensions = 512
|
| 11 |
|
| 12 |
# # 2. load model
|
| 13 |
# model = SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1", truncate_dim=dimensions)
|
|
@@ -44,8 +44,8 @@ class EmbeddingRequest(BaseModel):
|
|
| 44 |
|
| 45 |
|
| 46 |
models = {
|
| 47 |
-
"mxbai-embed-large":SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1"
|
| 48 |
-
"nomic-embed-text": SentenceTransformer("nomic-ai/nomic-embed-text-v1.5", trust_remote_code=True
|
| 49 |
}
|
| 50 |
|
| 51 |
|
|
|
|
| 7 |
import torch.nn.functional as F
|
| 8 |
|
| 9 |
# 1. Specify preffered dimensions
|
| 10 |
+
# dimensions = 512
|
| 11 |
|
| 12 |
# # 2. load model
|
| 13 |
# model = SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1", truncate_dim=dimensions)
|
|
|
|
| 44 |
|
| 45 |
|
| 46 |
models = {
|
| 47 |
+
"mxbai-embed-large":SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1"),
|
| 48 |
+
"nomic-embed-text": SentenceTransformer("nomic-ai/nomic-embed-text-v1.5", trust_remote_code=True)
|
| 49 |
}
|
| 50 |
|
| 51 |
|