Update app.py

app.py

@@ -1,17 +1,15 @@
 import os
-import io
 
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
 from PIL import Image, ImageDraw
 import traceback
 
-
 import gradio as gr
 
 import torch
 from docquery import pipeline
-from docquery.document import load_document, ImageDocumenta
+from docquery.document import load_document, ImageDocument
 from docquery.ocr_reader import get_ocr_reader
 
 
@@ -87,11 +85,8 @@ examples = [
         "statement.png",
         "What are net sales for 2020?",
     ],
-    [
-        "saleData.pdf",
-        "What is the highest sale amount of televsion in east region?",
-
-    ]
+    
+        
     #    [
     #        "docquery.png",
     #        "How many likes does the space have?",
@@ -106,8 +101,6 @@ 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",
-    "What is the highest sale amount of television in east region?": "SaleData.pdf",  # Add the PDF file here
-
 }
 
 
@@ -150,10 +143,6 @@ def process_upload(file):
             gr.update(visible=False, value=None),
             None,
         )
-    
-
-    
-
 
 
 colors = ["#64A087", "green", "black"]
@@ -236,7 +225,7 @@ CSS = """
     padding-bottom: 2px !important;
     padding-left: 8px !important;
     padding-right: 8px !important;
-        margin-top: 10px;
+	margin-top: 10px;
 }
 .gradio-container .gr-button-primary {
     background: linear-gradient(180deg, #CDF9BE 0%, #AFF497 100%);
@@ -302,15 +291,19 @@ gradio-app h2, .gradio-app h2 {
 """
 
 with gr.Blocks(css=CSS) as demo:
-    gr.Markdown("# Document Query Engine")
-    
+    gr.Markdown("# DocQuery: Document Query Engine")
+    gr.Markdown(
+        "DocQuery (created by [Impira](https://impira.com?utm_source=huggingface&utm_medium=referral&utm_campaign=docquery_space))"
+        " uses LayoutLMv1 fine-tuned on DocVQA, a document visual question"
+        " answering dataset, as well as SQuAD, which boosts its English-language comprehension."
+        " To use it, simply upload an image or PDF, type a question, and click 'submit', or "
+        " click one of the examples to load them."
+        " DocQuery is MIT-licensed and available on [Github](https://github.com/impira/docquery)."
+    )
 
     document = gr.Variable()
     example_question = gr.Textbox(visible=False)
     example_image = gr.Image(visible=False)
-    excel_upload = gr.File(label="Upload Excel", type="xlsx", elem_id="excel-upload-box")
-
-    excel_process_button = gr.Button("Process Excel", variant="primary", elem_id="excel-process-button")
 
     with gr.Row(equal_height=True):
         with gr.Column():
@@ -434,6 +427,5 @@ with gr.Blocks(css=CSS) as demo:
         outputs=[document, question, image, img_clear_button, output, output_text],
     )
 
-
 if __name__ == "__main__":
     demo.launch(enable_queue=False)