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

Update app.py

Browse files

changed embedding model

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ load_dotenv()
15
  assert os.getenv("TOGETHER_API_KEY"), "api key missing"
16
 
17
  # Use a sentence transformer for embeddings
18
- embedding_model = SentenceTransformer("BAAI/bge-base-en-v1.5") # Alternative: 'togethercomputer/m2-bert-80M-8k-retrieval'
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
+ 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