Namitg02 commited on
Commit
568a490
·
verified ·
1 Parent(s): e0f92b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -7,9 +7,9 @@ from langchain.text_splitter import RecursiveCharacterTextSplitter
7
  splitter = RecursiveCharacterTextSplitter(chunk_size=100, chunk_overlap=15,separators=["\n\n", "\n", " ", ""])
8
  docs = splitter.create_documents(str(dataset))
9
 
10
-
11
- from langchain_community.embeddings import HuggingFaceEmbeddings
12
- embedding_model = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
13
 
14
  from langchain_community.vectorstores import Chroma
15
  persist_directory = 'docs/chroma/'
 
7
  splitter = RecursiveCharacterTextSplitter(chunk_size=100, chunk_overlap=15,separators=["\n\n", "\n", " ", ""])
8
  docs = splitter.create_documents(str(dataset))
9
 
10
+ from sentence_transformers import SentenceTransformer
11
+ #from langchain_community.embeddings import HuggingFaceEmbeddings
12
+ embedding_model = SentenceTransformer(model_name="all-MiniLM-L6-v2")
13
 
14
  from langchain_community.vectorstores import Chroma
15
  persist_directory = 'docs/chroma/'