moriire commited on
Commit
8438871
·
verified ·
1 Parent(s): b8a733e

Update app/injest.py

Browse files
Files changed (1) hide show
  1. app/injest.py +1 -1
app/injest.py CHANGED
@@ -17,7 +17,7 @@ def create_vector_db():
17
  chunk_overlap=50)
18
  texts = text_splitter.split_documents(documents)
19
 
20
- embeddings = HuggingFaceEmbeddings(model_name='sentence-transformers/all-MiniLM-L6-v2',
21
  model_kwargs={'device': 'cpu'})
22
 
23
  db = FAISS.from_documents(texts, embeddings)
 
17
  chunk_overlap=50)
18
  texts = text_splitter.split_documents(documents)
19
 
20
+ embeddings = HuggingFaceEmbeddings(model_name='Qwen/Qwen1.5-0.5B-Chat-GGUF',
21
  model_kwargs={'device': 'cpu'})
22
 
23
  db = FAISS.from_documents(texts, embeddings)