Spaces:
Sleeping
Sleeping
Update rag_retriever.py
Browse files- rag_retriever.py +1 -1
rag_retriever.py
CHANGED
@@ -17,7 +17,7 @@ def initialize_llm():
|
|
17 |
load_dotenv()
|
18 |
HF_TOKEN = os.getenv("HUGGINGFACE_API_KEY")
|
19 |
|
20 |
-
Settings.llm = HuggingFaceInferenceAPI(model_name="
|
21 |
Settings.embed_model = HuggingFaceEmbedding(model_name="BAAI/bge-small-en-v1.5")
|
22 |
|
23 |
|
|
|
17 |
load_dotenv()
|
18 |
HF_TOKEN = os.getenv("HUGGINGFACE_API_KEY")
|
19 |
|
20 |
+
Settings.llm = HuggingFaceInferenceAPI(model_name="HuggingFaceH4/zephyr-7b-alpha", token=HUGGINGFACE_API_KEY)
|
21 |
Settings.embed_model = HuggingFaceEmbedding(model_name="BAAI/bge-small-en-v1.5")
|
22 |
|
23 |
|