jonathanjordan21 commited on
Commit
ea4f0af
·
verified ·
1 Parent(s): b11de04

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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", truncate_dim=dimensions),
48
- "nomic-embed-text": SentenceTransformer("nomic-ai/nomic-embed-text-v1.5", trust_remote_code=True, truncate_dim=dimensions)
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