update app.py

app.py

@@ -112,40 +112,40 @@ class PdfReader(object):
 
 examples = [
     [
-        "invoice.jpg",
-        "发票号码是多少?",
+        "budget_form.png",
+        "What is the total actual and/or obligated expenses of ECG Center?"
     ],
     [
-        "resume.png",
-        "五百丁本次想要担任的是什么职位?",
+        "medical_bill_2.png",
+        "患者さんは何でお金を払いますか。"
     ],
     [
-        "custom_declaration_form.png",
-        "在哪个口岸进口？"
+        "receipt.png",
+        "เบอร์โทรร้านอะไรคะ"
     ],
     [
-        "medical_bill_1.png",
-        "票据的具体名称是什么？"
+        "poster.png",
+        "Which gift idea needs a printer?"
     ],
     [
-        "budget_form.png",
-        "What is the total actual and/or obligated expenses of ECG Center?"
+        "resume.png",
+        "五百丁本次想要担任的是什么职位?",
     ],
     [
-        "website_design_guide.jpeg",
-        "Which quality component has the icon of a pen in it?"
+        "custom_declaration_form.png",
+        "在哪个口岸进口？"
     ],
     [
-        "poster.png",
-        "Which gift idea needs a printer?"
+        "invoice.jpg",
+        "发票号码是多少?",
     ],
     [
-        "medical_bill_2.png",
-        "患者さんは何でお金を払いますか。"
+        "medical_bill_1.png",
+        "票据的具体名称是什么？"
     ],
     [
-        "receipt.png",
-        "เบอร์โทรร้านอะไรคะ"
+        "website_design_guide.jpeg",
+        "Which quality component has the icon of a pen in it?"
     ],
 ]
 
@@ -250,10 +250,10 @@ def process_prompt(prompt, document, lang="ch"):
     )
 
 
-def load_example_document(img, prompt):
+def load_example_document(img, prompt, lang="ch"):
     if img is not None:
         document = prompt_files[prompt]
-        preview, answer, answer_text = process_prompt(prompt, document)
+        preview, answer, answer_text = process_prompt(prompt, document, lang)
         return document, prompt, preview, gr.update(visible=True), answer, answer_text
     else:
         return None, None, None, gr.update(visible=False), None, None
@@ -355,9 +355,8 @@ gradio-app h2, .gradio-app h2 {
 with gr.Blocks(css=CSS) as demo:
     gr.HTML(read_content("header.html"))
     gr.Markdown(
-        f" ⚡DocPrompt⚡ is a Document Prompt Engine Powered by Baidu Wenxin Document Intelligence. 🚀\n"
-        f" To use it, simply upload an image or PDF, type the prompt, and click 'submit', or "
-        f" click one of the examples to load them."
+        f" ⚡DocPrompt⚡ is a Document Prompt Engine uses ERNIE-LayoutX as the backbone model.\n"
+        f" The engine is powered by Baidu Wenxin Document Intelligence Team 🚀 and is ability for multilingual documents information extraction and question ansering.\n"
     )
 
     document = gr.Variable()
@@ -463,7 +462,7 @@ with gr.Blocks(css=CSS) as demo:
 
     prompt.submit(
         fn=process_prompt,
-        inputs=[prompt, document],
+        inputs=[prompt, document, ocr_lang],
         outputs=[image, output, output_text],
     )
 
@@ -481,11 +480,10 @@ with gr.Blocks(css=CSS) as demo:
 
     example_image.change(
         fn=load_example_document,
-        inputs=[example_image, example_prompt],
+        inputs=[example_image, example_prompt, ocr_lang],
         outputs=[document, prompt, image, img_clear_button, output, output_text],
     )
 
-    gr.Image('./paddlenlp-preview.jpeg')
     gr.Markdown("[![Stargazers repo roster for @PaddlePaddle/PaddleNLP](https://reporoster.com/stars/PaddlePaddle/PaddleNLP)](https://github.com/PaddlePaddle/PaddleNLP)")
     gr.HTML(read_content("footer.html"))
 

header.html

@@ -20,11 +20,7 @@
         margin-bottom: 10px;
         justify-content: center;
       ">
-    <a href="https://github.com/PaddlePaddle/PaddleNLP"><h1 style="font-weight: 900; margin-bottom: 7px;">
-      DocPrompt 🚀 
+    <a href="https://github.com/PaddlePaddle/PaddleNLP"><h1 style="font-weight: 900; align-items: center; margin-bottom: 7px;">
+      ERNIE-LayoutX DocPrompt Engine
     </h1></a>
-    </div> 
-    <!-- <p style="margin-bottom: 10px; font-weight: 900; font-size: 100%">
-      ⚡DocPrompt⚡ is a Document Prompt Engine Powered by <br> Document Intelligence Technology in Baidu Wenxin<br>. 🚀 To use it, simply upload an image or PDF, type the prompt, and click 'submit', or click one of the examples to load them.
-    </p>
-  </div> -->
+    </div>