dwh1142 commited on
Commit
efc9dcd
1 Parent(s): 1fdb529

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -11
app.py CHANGED
@@ -5,20 +5,21 @@ from transformers import pipeline
5
  pipe = pipeline(model="deepset/roberta-base-squad2")
6
 
7
  def predict(file_obj, question):
8
- if file_obj:
9
- docText = []
10
- reader = PdfReader(file_obj)
11
- for page in reader.pages:
12
- txt = page.extract_text()
13
- docText.append(txt)
14
 
15
- qaInput = {
16
- 'question': question,
17
- 'context': docText
18
- }
19
 
20
- return pipe(qaInput)
 
 
21
 
 
 
 
 
 
 
22
 
23
  demo = gr.Interface(
24
  fn=predict,
 
5
  pipe = pipeline(model="deepset/roberta-base-squad2")
6
 
7
  def predict(file_obj, question):
8
+ docText = []
 
 
 
 
 
9
 
10
+ with open(file_obj, 'r') as f:
11
+ reader = PdfReader(f)
 
 
12
 
13
+ for page in reader.pages:
14
+ txt = page.extract_text()
15
+ docText.append(txt)
16
 
17
+ qaInput = {
18
+ 'question': question,
19
+ 'context': docText
20
+ }
21
+
22
+ return pipe(qaInput)
23
 
24
  demo = gr.Interface(
25
  fn=predict,