harshpatel080503 commited on
Commit
ea3695b
·
verified ·
1 Parent(s): 541484c

Update rag_chain.py

Browse files
Files changed (1) hide show
  1. rag_chain.py +4 -1
rag_chain.py CHANGED
@@ -23,7 +23,10 @@ os.environ["OPENAI_API_KEY"] = openai_token
23
 
24
  # Hugging Face Embeddings
25
  os.environ['HF_HOME'] = 'Embedding Models'
26
- embedding = HuggingFaceEmbeddings(model_name='sentence-transformers/all-MiniLM-L6-v2')
 
 
 
27
 
28
  # OpenRouter LLM (Meta LLaMA 3.3)
29
  llm = ChatOpenAI(
 
23
 
24
  # Hugging Face Embeddings
25
  os.environ['HF_HOME'] = 'Embedding Models'
26
+ embedding = HuggingFaceEmbeddings(
27
+ model_name="Embedding Models/hub/models--sentence-transformers--all-MiniLM-L6-v2",
28
+ model_kwargs={"local_files_only": True}
29
+ )
30
 
31
  # OpenRouter LLM (Meta LLaMA 3.3)
32
  llm = ChatOpenAI(