Update app.py
Browse files
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.
|
| 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
|