Penality commited on
Commit
4468b37
·
verified ·
1 Parent(s): 92a9c14

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -15,7 +15,14 @@ load_dotenv()
15
  assert os.getenv("TOGETHER_API_KEY"), "api key missing"
16
 
17
  # Use a sentence transformer for embeddings
18
- embedding_model = SentenceTransformer("togethercomputer/m2-bert-80M-8k-retrieval") # Alternative: 'togethercomputer/m2-bert-80M-8k-retrieval' 'BAAI/bge-base-en-v1.5'
 
 
 
 
 
 
 
19
  embedding_dim = 768 # Adjust according to model
20
 
21
  # Initialize FAISS index
 
15
  assert os.getenv("TOGETHER_API_KEY"), "api key missing"
16
 
17
  # Use a sentence transformer for embeddings
18
+ #'BAAI/bge-base-en-v1.5'
19
+ # embedding_model = SentenceTransformer("BAAI/bge-base-en-v1.5")
20
+
21
+ # 'togethercomputer/m2-bert-80M-8k-retrieval'
22
+ embedding_model = SentenceTransformer(
23
+ "togethercomputer/m2-bert-80M-8k-retrieval",
24
+ trust_remote_code=True # Allow remote code execution
25
+ )
26
  embedding_dim = 768 # Adjust according to model
27
 
28
  # Initialize FAISS index