Use PDF for statement

app.py

@@ -84,6 +84,10 @@ examples = [
     ],
 ]
 
+question_files = {
+    "What are net sales for 2020?": "statement.pdf",
+}
+
 
 def process_path(path):
     if path:
@@ -162,7 +166,11 @@ def process_question(question, document, model=list(CHECKPOINTS.keys())[0]):
 
 def load_example_document(img, question, model):
     if img is not None:
-        document = ImageDocument(Image.fromarray(img))
+        if question in question_files:
+            print("using document")
+            document = load_document(question_files[question])
+        else:
+            document = ImageDocument(Image.fromarray(img))
         preview, answer, answer_text = process_question(question, document, model)
         return document, question, preview, gr.update(visible=True), answer, answer_text
     else: