Update src/backend/embed_models.py
Browse files
src/backend/embed_models.py
CHANGED
|
@@ -5,7 +5,8 @@ import os
|
|
| 5 |
load_dotenv()
|
| 6 |
from huggingface_hub import login
|
| 7 |
|
| 8 |
-
login(token=os.getenv("HUGGINGFACE_HUB_TOKEN"))
|
|
|
|
| 9 |
|
| 10 |
dense_model = TextEmbedding("sentence-transformers/all-MiniLM-L6-v2")
|
| 11 |
colbert_model = LateInteractionTextEmbedding("colbert-ir/colbertv2.0")
|
|
|
|
| 5 |
load_dotenv()
|
| 6 |
from huggingface_hub import login
|
| 7 |
|
| 8 |
+
#login(token=os.getenv("HUGGINGFACE_HUB_TOKEN"))
|
| 9 |
+
hf_token = os.getenv("HUGGINGFACE_HUB_TOKEN")
|
| 10 |
|
| 11 |
dense_model = TextEmbedding("sentence-transformers/all-MiniLM-L6-v2")
|
| 12 |
colbert_model = LateInteractionTextEmbedding("colbert-ir/colbertv2.0")
|