AyoubChLin commited on
Commit
27901ec
·
verified ·
1 Parent(s): 94a26cf

Update embedding.py

Browse files
Files changed (1) hide show
  1. embedding.py +6 -3
embedding.py CHANGED
@@ -1,8 +1,11 @@
1
- from langchain.embeddings import HuggingFaceInferenceAPIEmbeddings
 
 
2
  from dotenv import load_dotenv
3
  import os
4
- load_dotenv()
5
 
 
 
6
  embeddings = HuggingFaceInferenceAPIEmbeddings(
7
- api_key=os.getenv("HUGGINGFACEHUB_API_TOKEN"), model_name="sentence-transformers/all-MiniLM-L6-v2"
8
  )
 
1
+ from langchain_community.embeddings import (
2
+ HuggingFaceInferenceAPIEmbeddings,
3
+ )
4
  from dotenv import load_dotenv
5
  import os
 
6
 
7
+ load_dotenv()
8
+ api_key = os.getenv("HUGGINGFACEHUB_API_TOKEN")
9
  embeddings = HuggingFaceInferenceAPIEmbeddings(
10
+ api_key=api_key, model_name="sentence-transformers/all-MiniLM-L6-v2"
11
  )