Chris4K commited on
Commit
05727e9
·
1 Parent(s): cae3cb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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