gabejavitt commited on
Commit
15dfb45
·
verified ·
1 Parent(s): 1e58285

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -850,7 +850,7 @@ def scrape_and_retrieve(url: str, query: str) -> str:
850
  docs = [Document(page_content=c, metadata={"source": url}) for c in chunks]
851
 
852
  db = FAISS.from_documents(docs, global_embeddings)
853
- retriever = db.as_retriever(search_kwargs={"k": 7})
854
  retrieved = retriever.invoke(query)
855
 
856
  if not retrieved:
 
850
  docs = [Document(page_content=c, metadata={"source": url}) for c in chunks]
851
 
852
  db = FAISS.from_documents(docs, global_embeddings)
853
+ retriever = db.as_retriever(search_kwargs={"k": 5})
854
  retrieved = retriever.invoke(query)
855
 
856
  if not retrieved: