jinysun commited on
Commit
ad5aba5
·
verified ·
1 Parent(s): ef91a27

Update tool/rag.py

Browse files
Files changed (1) hide show
  1. tool/rag.py +1 -1
tool/rag.py CHANGED
@@ -32,7 +32,7 @@ device = f'cuda:{cuda.current_device()}' if cuda.is_available() else 'cpu'
32
  from langchain_openai import OpenAIEmbeddings
33
  embeddings = OpenAIEmbeddings(api_key='sk-itPrztYm9F6XZZpsBMJB9O7Vq0pYUABVVBSoThuBxEGTnDik',
34
  base_url="https://www.dmxapi.com/v1")
35
- vectorstore=FAISS.load_local(r"tool/rag", embeddings,allow_dangerous_deserialization =True)
36
 
37
 
38
  template = """
 
32
  from langchain_openai import OpenAIEmbeddings
33
  embeddings = OpenAIEmbeddings(api_key='sk-itPrztYm9F6XZZpsBMJB9O7Vq0pYUABVVBSoThuBxEGTnDik',
34
  base_url="https://www.dmxapi.com/v1")
35
+ vectorstore=FAISS.load_local(r"https://huggingface.co/spaces/jinysun/TeLLAgent/blob/main/tool/rag", embeddings,allow_dangerous_deserialization =True)
36
 
37
 
38
  template = """