redfernstech commited on
Commit
26fe4df
·
verified ·
1 Parent(s): 45de4f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -40,8 +40,8 @@ text_splitter = RecursiveCharacterTextSplitter(chunk_size=10000, chunk_overlap=1
40
  context = "\n\n".join(str(p.page_content) for p in pages)
41
  texts = text_splitter.split_text(context)
42
  #embeddings = GoogleGenerativeAIEmbeddings(model="models/embedding-001",google_api_key=GOOGLE_API_KEY)
43
- sentence_transformer_model = SentenceTransformer('BAAI/bge-small-en-v1.5')
44
- embeddings = HuggingFaceEmbeddings(model=sentence_transformer_model)
45
  vector_index = Chroma.from_texts(texts, embeddings).as_retriever(search_kwargs={"k":10})
46
  qa_chain = RetrievalQA.from_chain_type(
47
  model,
 
40
  context = "\n\n".join(str(p.page_content) for p in pages)
41
  texts = text_splitter.split_text(context)
42
  #embeddings = GoogleGenerativeAIEmbeddings(model="models/embedding-001",google_api_key=GOOGLE_API_KEY)
43
+ #sentence_transformer_model = SentenceTransformer()
44
+ embeddings = HuggingFaceEmbeddings(model_name='BAAI/bge-small-en-v1.5')
45
  vector_index = Chroma.from_texts(texts, embeddings).as_retriever(search_kwargs={"k":10})
46
  qa_chain = RetrievalQA.from_chain_type(
47
  model,