Vallabhpatil777 commited on
Commit
77861f1
·
verified ·
1 Parent(s): ff58d3c

Update src/backend/embed_models.py

Browse files
Files changed (1) hide show
  1. src/backend/embed_models.py +2 -1
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")