Spaces:
Sleeping
Sleeping
Update app.py
Browse fileschanged embedding model
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("
|
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
|