yanqiang commited on
Commit
1c6e8fd
1 Parent(s): 8941872

feature@init

Browse files
Files changed (1) hide show
  1. tests/test_langchain.py +1 -1
tests/test_langchain.py CHANGED
@@ -4,7 +4,7 @@ from langchain.document_loaders import UnstructuredFileLoader
4
  from langchain.embeddings.huggingface import HuggingFaceEmbeddings
5
  from langchain.vectorstores import FAISS
6
 
7
- embedding_model_name = '/home/searchgpt/pretrained_models/ernie-gram-zh'
8
  docs_path = 'docs'
9
  embeddings = HuggingFaceEmbeddings(model_name=embedding_model_name)
10
 
 
4
  from langchain.embeddings.huggingface import HuggingFaceEmbeddings
5
  from langchain.vectorstores import FAISS
6
 
7
+ embedding_model_name = 'pretrained_models/ernie-gram-zh'
8
  docs_path = 'docs'
9
  embeddings = HuggingFaceEmbeddings(model_name=embedding_model_name)
10