CatoG commited on
Commit
17f8afb
·
verified ·
1 Parent(s): 0c0b9d5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -144,7 +144,7 @@ def retriever_qa(file, query, model_choice, max_tokens, temperature, embedding_m
144
  retriever_obj = retriever(file, int(chunk_size), int(chunk_overlap), embedding_model)
145
 
146
  # Get relevant documents
147
- docs = retriever_obj.get_relevant_documents(query)
148
  context = "\n\n".join(doc.page_content for doc in docs)
149
 
150
  # Create prompt
 
144
  retriever_obj = retriever(file, int(chunk_size), int(chunk_overlap), embedding_model)
145
 
146
  # Get relevant documents
147
+ docs = retriever_obj.invoke(query)
148
  context = "\n\n".join(doc.page_content for doc in docs)
149
 
150
  # Create prompt