bluuebunny commited on
Commit
687db51
·
1 Parent(s): 04f8232

changed model to "mixedbread-ai/mxbai-embed-large-v1"

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import torch # For gpu
7
  device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
8
 
9
  # Load a pretrained Sentence Transformer model and move it to the appropriate device
10
- model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2', trust_remote_code=True)
11
  model = model.to(device)
12
 
13
  # Function that does the embedding
 
7
  device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
8
 
9
  # Load a pretrained Sentence Transformer model and move it to the appropriate device
10
+ model = SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1")
11
  model = model.to(device)
12
 
13
  # Function that does the embedding