Update app.py

app.py

@@ -21,9 +21,10 @@ def ensure_list(x):
 
 
 CHECKPOINTS = {
-    "LayoutLMv1 🦉": "impira/layoutlm-document-qa",
-    "LayoutLMv1 for Invoices 💸": "impira/layoutlm-invoices",
-    "Donut 🍩": "naver-clova-ix/donut-base-finetuned-docvqa",
+    "CQI Kleber 🤖": "cloudqi/CQI_Visual_Question_Awnser_PT_v0",
+    "LayoutLMv1 Padrão": "impira/layoutlm-document-qa",
+    "LayoutLMv1 para Faturas 💸": "impira/layoutlm-invoices",
+    "Donut": "naver-clova-ix/donut-base-finetuned-docvqa",
 }
 
 PIPELINES = {}
@@ -74,15 +75,15 @@ def normalize_bbox(box, width, height, padding=0.005):
 examples = [
     [
         "invoice.png",
-        "What is the invoice number?",
+        "Qual é o número da fatura?",
     ],
     [
         "contract.jpeg",
-        "What is the purchase amount?",
+        "Qual é o valor da compra?",
     ],
     [
         "statement.png",
-        "What are net sales for 2020?",
+        "Quais são as vendas líquidas para 2020?",
     ],
     #    [
     #        "docquery.png",
@@ -95,9 +96,9 @@ examples = [
 ]
 
 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",
+    "Quais são as vendas líquidas para 2020?": "statement.pdf",
+    "Quantas curtidas tem o espaço?": "https://huggingface.co/spaces/impira/docquery",
+    "Qual é o título do post número 5?": "https://news.ycombinator.com",
 }
 
 
@@ -288,15 +289,7 @@ gradio-app h2, .gradio-app h2 {
 """
 
 with gr.Blocks(css=CSS) as demo:
-    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)."
-    )
+    gr.Markdown("# Mecanismo de Consulta de Documentos")
 
     document = gr.Variable()
     example_question = gr.Textbox(visible=False)
@@ -305,9 +298,9 @@ with gr.Blocks(css=CSS) as demo:
     with gr.Row(equal_height=True):
         with gr.Column():
             with gr.Row():
-                gr.Markdown("## 1. Select a file", elem_id="select-a-file")
+                gr.Markdown("## 1. Selecione um arquivo", elem_id="select-a-file")
                 img_clear_button = gr.Button(
-                    "Clear", variant="secondary", elem_id="file-clear", visible=False
+                    "Limpar", variant="secondary", elem_id="file-clear", visible=False
                 )
             image = gr.Gallery(visible=False)
             with gr.Row(equal_height=True):
@@ -336,10 +329,10 @@ with gr.Blocks(css=CSS) as demo:
             )
 
         with gr.Column() as col:
-            gr.Markdown("## 2. Ask a question")
+            gr.Markdown("## 2. Faça uma pergunta")
             question = gr.Textbox(
                 label="Question",
-                placeholder="e.g. What is the invoice number?",
+                placeholder="e.g. Qual é o número da fatura?",
                 lines=1,
                 max_lines=1,
             )
@@ -352,11 +345,11 @@ with gr.Blocks(css=CSS) as demo:
             with gr.Row():
                 clear_button = gr.Button("Clear", variant="secondary")
                 submit_button = gr.Button(
-                    "Submit", variant="primary", elem_id="submit-button"
+                    "Enviar", variant="primary", elem_id="submit-button"
                 )
             with gr.Column():
                 output_text = gr.Textbox(
-                    label="Top Answer", visible=False, elem_id="answer"
+                    label="Resposta principal", visible=False, elem_id="answer"
                 )
                 output = gr.JSON(label="Output", visible=False)