xavierbarbier commited on
Commit
c673c9d
·
verified ·
1 Parent(s): bc6cbab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -15,9 +15,13 @@ p = pipeline(
15
  def qa(question: str, doc: str) -> str:
16
  reader = PdfReader(doc)
17
 
18
- text = reader.pages[0]
 
 
19
 
20
- #img = convert_from_path(doc)[0]
 
 
21
  #output = p(img, question)
22
  return text
23
 
 
15
  def qa(question: str, doc: str) -> str:
16
  reader = PdfReader(doc)
17
 
18
+ text = []
19
+ for p in np.arange(0, len(reader.pages), 1):
20
+ page = reader.pages[int(p)]
21
 
22
+ # extracting text from page
23
+ text.append(page.extract_text())
24
+ text = ' '.join(text)
25
  #output = p(img, question)
26
  return text
27