Update app.py

app.py

@@ -72,26 +72,6 @@ def normalize_bbox(box, width, height, padding=0.005):
 
 
 examples = [
-    [
-        "invoice.png",
-        "What is the invoice number?",
-    ],
-    [
-        "contract.jpeg",
-        "What is the purchase amount?",
-    ],
-    [
-        "statement.png",
-        "What are net sales for 2020?",
-    ],
-    #    [
-    #        "docquery.png",
-    #        "How many likes does the space have?",
-    #    ],
-    #    [
-    #        "hacker_news.png",
-    #        "What is the title of post number 5?",
-    #    ],
 ]
 
 question_files = {
@@ -145,7 +125,7 @@ def process_upload(file):
 colors = ["#64A087", "green", "black"]
 
 
-def process_question(question, document, model=list(CHECKPOINTS.keys())[0]):
+def process_question(question = "What is the purchase amount?", document, model=list(CHECKPOINTS.keys())[0]):
     if not question or document is None:
         return None, None, None
 
@@ -183,10 +163,7 @@ def process_question(question, document, model=list(CHECKPOINTS.keys())[0]):
 
 def load_example_document(img, question, model):
     if img is not None:
-        if question in question_files:
-            document = load_document(question_files[question])
-        else:
-            document = ImageDocument(Image.fromarray(img), get_ocr_reader())
+        document = ImageDocument(Image.fromarray(img), get_ocr_reader())
         preview, answer, answer_text = process_question(question, document, model)
         return document, question, preview, gr.update(visible=True), answer, answer_text
     else: