Update app.py
Browse files
app.py
CHANGED
@@ -26,6 +26,8 @@ vdocuments = text_splitter.split_documents(documents)
|
|
26 |
# Create Chroma vector store for API embeddings
|
27 |
api_db = Chroma.from_documents(vdocuments, api_hf_embeddings, collection_name="api-collection")
|
28 |
|
|
|
|
|
29 |
# Define the PDF retrieval function
|
30 |
def pdf_retrieval(query):
|
31 |
# Run the query through the retriever
|
|
|
26 |
# Create Chroma vector store for API embeddings
|
27 |
api_db = Chroma.from_documents(vdocuments, api_hf_embeddings, collection_name="api-collection")
|
28 |
|
29 |
+
print(api_db.similarity_search("What is react"))
|
30 |
+
|
31 |
# Define the PDF retrieval function
|
32 |
def pdf_retrieval(query):
|
33 |
# Run the query through the retriever
|