Update app.py

app.py

@@ -1,5 +1,28 @@
 import gradio as gr
+from pypdf import PdfReader
+from transformers import pipeline
 
-qa = gr.load("impira/layoutlm-document-qa", src="models")
+pipe = pipeline(model="deepset/roberta-base-squad2")
 
-qa.launch()
+def predict(file, question):
+    docText = []
+    reader = PdfReader(file)
+    for page in reader.pages:
+        txt = page.extract_text()
+        docText.append(txt)
+
+    qaInput = {
+    'question': question,
+    'context': docText
+    }
+    
+    return pipe(qaInput)
+
+
+demo = gr.Interface(
+  fn=predict,
+  inputs=[gr.File(), "text"]
+  outputs="text",
+)
+
+demo.launch()