Spaces:
Build error
Build error
Update app.py
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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
|