Spaces:

Pranit4u
/

portfolio-rag

Runtime error

Pranit4u commited on Apr 21

Commit

ac97fe2

•

1 Parent(s): 86e4032

Update get_embedding_function.py

Files changed (1) hide show

get_embedding_function.py CHANGED Viewed

@@ -3,6 +3,6 @@ from langchain_community.embeddings import HuggingFaceInferenceAPIEmbeddings
 def get_embedding_function():
     embeddings = HuggingFaceInferenceAPIEmbeddings(
-        api_key=os.environ.get("HUGGINGFACE_API_KEY"), model_name="BAAI/bge-large-en-v1.5"
     )
     return embeddings

 def get_embedding_function():
     embeddings = HuggingFaceInferenceAPIEmbeddings(
+        api_key=os.environ.get("HUGGINGFACE_API_KEY"), model_name="BAAI/bge-small-en-v1.5"
     )
     return embeddings