Update app.py
Browse files
app.py
CHANGED
@@ -21,7 +21,8 @@ def pdf_changes(pdf_doc):
|
|
21 |
texts = text_splitter.split_documents(documents)
|
22 |
db = Chroma.from_documents(texts, embeddings)
|
23 |
retriever = db.as_retriever()
|
24 |
-
global qa
|
|
|
25 |
return "Ready"
|
26 |
|
27 |
def infer(question):
|
|
|
21 |
texts = text_splitter.split_documents(documents)
|
22 |
db = Chroma.from_documents(texts, embeddings)
|
23 |
retriever = db.as_retriever()
|
24 |
+
global qa
|
25 |
+
qa = RetrievalQA.from_chain_type(llm=flan_ul2, chain_type="stuff", retriever=retriever, return_source_documents=True)
|
26 |
return "Ready"
|
27 |
|
28 |
def infer(question):
|