chatbytes commited on
Commit
d17191b
·
verified ·
1 Parent(s): 6579def

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -6
app.py CHANGED
@@ -32,17 +32,16 @@ def text_extract(file):
32
  page = pdf_reader.pages[page_num]
33
  text += page.extract_text()
34
  text_splitter=text_splitter_function(text);
35
- result=response_from_bot(text_splitter)
36
- return result
37
- def response_from_bot(text_splitter):
38
  db = FAISS.from_texts(text_splitter, embeddings);
39
  retriever = db.as_retriever(search_type="similarity", search_kwargs={"k": 2})
40
  llm=GooglePalm(google_api_key=google_api)
41
  qa = RetrievalQA.from_chain_type(
42
- llm=llm, chain_type="stuff", retriever=retriever, return_source_documents=True
43
- )
44
  result=qa.invoke("where is tajmahal")
45
- return result['result'];
 
 
46
 
47
 
48
 
 
32
  page = pdf_reader.pages[page_num]
33
  text += page.extract_text()
34
  text_splitter=text_splitter_function(text);
 
 
 
35
  db = FAISS.from_texts(text_splitter, embeddings);
36
  retriever = db.as_retriever(search_type="similarity", search_kwargs={"k": 2})
37
  llm=GooglePalm(google_api_key=google_api)
38
  qa = RetrievalQA.from_chain_type(
39
+ llm=llm, chain_type="stuff", retriever=retriever, return_source_documents=True
40
+ )
41
  result=qa.invoke("where is tajmahal")
42
+
43
+ return result['result']
44
+
45
 
46
 
47