Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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(
|
44 |
-
embeddings = HuggingFaceEmbeddings(
|
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,
|