dnzblgn commited on
Commit
c878a32
·
verified ·
1 Parent(s): 36b33e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -33,7 +33,10 @@ def read_file(file_path):
33
  def create_db_from_text(text):
34
  text_splitter = RecursiveCharacterTextSplitter(chunk_size=1024, chunk_overlap=64)
35
  splits = text_splitter.create_documents([text])
36
- embeddings = HuggingFaceEmbeddings()
 
 
 
37
  vector_db = FAISS.from_documents(splits, embeddings)
38
  return vector_db
39
 
 
33
  def create_db_from_text(text):
34
  text_splitter = RecursiveCharacterTextSplitter(chunk_size=1024, chunk_overlap=64)
35
  splits = text_splitter.create_documents([text])
36
+
37
+ # Specify an explicit model for embeddings
38
+ embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
39
+
40
  vector_db = FAISS.from_documents(splits, embeddings)
41
  return vector_db
42