Update app.py

app.py

@@ -151,6 +151,18 @@ prompt_files = {
     "เบอร์โทรร้านอะไรคะ?": "receipt.png",
 }
 
+lang_map = {
+    "invoice.jpg": "ch",
+    "resume.png": "ch",
+    "custom_declaration_form.png": "ch",
+    "medical_bill_1.png": "ch",
+    "budget_form.png": "en",
+    "website_design_guide.jpeg": "en",
+    "poster.png": "en",
+    "medical_bill_2.jpg": "ch",
+    "receipt.png": "en"
+}
+
 
 def load_document(path):
     if path.startswith("http://") or path.startswith("https://"):
@@ -267,9 +279,10 @@ def process_prompt(prompt, document, lang="ch"):
     )
 
 
-def load_example_document(img, prompt, lang="ch"):
+def load_example_document(img, prompt):
     if img is not None:
         document = prompt_files[prompt]
+        lang = lang_map[document]
         preview, answer, answer_text = process_prompt(prompt, document, lang)
         return document, prompt, preview, gr.update(visible=True), answer, answer_text
     else:
@@ -493,7 +506,7 @@ with gr.Blocks(css=CSS) as demo:
 
     example_image.change(
         fn=load_example_document,
-        inputs=[example_image, example_prompt, ocr_lang],
+        inputs=[example_image, example_prompt],
         outputs=[document, prompt, image, img_clear_button, output, output_text],
     )