Update app.py

app.py

@@ -22,6 +22,7 @@ def ensure_list(x):
 
 CHECKPOINTS = {
     "LayoutLMv1": "impira/layoutlm-document-qa",
+    "LayoutLMv1 for Invoices 💸": "impira/layoutlm-invoices",
 }
 
 PIPELINES = {}
@@ -68,7 +69,34 @@ def normalize_bbox(box, width, height, padding=0.005):
         max_y = min(max_y + padding, 1)
     return [min_x * width, min_y * height, max_x * width, max_y * height]
 
-
+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 = {
+    "What are net sales for 2020?": "statement.pdf",
+    "How many likes does the space have?": "https://huggingface.co/spaces/impira/docquery",
+    "What is the title of post number 5?": "https://news.ycombinator.com",
+}
 
 
 def process_path(path):
@@ -297,6 +325,10 @@ with gr.Blocks(css=CSS) as demo:
                     )
             gr.Markdown("— or —")
             upload = gr.File(label=None, interactive=True, elem_id="short-upload-box")
+            gr.Examples(
+                examples=examples,
+                inputs=[example_image, example_question],
+            )
 
         with gr.Column() as col:
             gr.Markdown("## 2. Ask a question")