Update app.py

app.py

@@ -5,8 +5,9 @@ from pathlib import Path
 from typing import List, Union
 
 import gradio as gr
-import rapidocr
 from omegaconf import OmegaConf
+
+import rapidocr
 from rapidocr import (
     EngineType,
     LangCls,
@@ -108,16 +109,16 @@ def create_examples() -> List[List[Union[str, float]]]:
             2000,
             EngineType.ONNXRUNTIME,
             LangDet.CH,
-            ModelType.MOBILE,
-            OCRVersion.PPOCRV5,
+            ModelType.SMALL,
+            OCRVersion.PPOCRV6,
             EngineType.ONNXRUNTIME,
             LangCls.CH,
             ModelType.MOBILE,
             OCRVersion.PPOCRV4,
             EngineType.ONNXRUNTIME,
             LangRec.CH,
-            ModelType.MOBILE,
-            OCRVersion.PPOCRV5,
+            ModelType.SMALL,
+            OCRVersion.PPOCRV6,
             "No",
             ["use_det", "use_cls", "use_rec"],
         ],
@@ -129,16 +130,16 @@ def create_examples() -> List[List[Union[str, float]]]:
             2000,
             EngineType.ONNXRUNTIME,
             LangDet.CH,
-            ModelType.MOBILE,
-            OCRVersion.PPOCRV4,
+            ModelType.SMALL,
+            OCRVersion.PPOCRV6,
             EngineType.ONNXRUNTIME,
             LangCls.CH,
             ModelType.MOBILE,
             OCRVersion.PPOCRV4,
             EngineType.ONNXRUNTIME,
             LangRec.CH,
-            ModelType.MOBILE,
-            OCRVersion.PPOCRV4,
+            ModelType.SMALL,
+            OCRVersion.PPOCRV6,
             "No",
             ["use_det", "use_cls", "use_rec"],
         ],
@@ -150,16 +151,16 @@ def create_examples() -> List[List[Union[str, float]]]:
             2000,
             EngineType.ONNXRUNTIME,
             LangDet.CH,
-            ModelType.MOBILE,
-            OCRVersion.PPOCRV5,
+            ModelType.SMALL,
+            OCRVersion.PPOCRV6,
             EngineType.ONNXRUNTIME,
             LangCls.CH,
             ModelType.MOBILE,
             OCRVersion.PPOCRV4,
             EngineType.ONNXRUNTIME,
             LangRec.CH,
-            ModelType.MOBILE,
-            OCRVersion.PPOCRV5,
+            ModelType.SMALL,
+            OCRVersion.PPOCRV6,
             "No",
             ["use_det", "use_cls", "use_rec"],
         ],
@@ -170,17 +171,17 @@ def create_examples() -> List[List[Union[str, float]]]:
             1.6,
             2000,
             EngineType.ONNXRUNTIME,
-            LangDet.MULTI,
-            ModelType.MOBILE,
-            OCRVersion.PPOCRV4,
+            LangDet.CH,
+            ModelType.SMALL,
+            OCRVersion.PPOCRV6,
             EngineType.ONNXRUNTIME,
             LangCls.CH,
             ModelType.MOBILE,
             OCRVersion.PPOCRV4,
             EngineType.ONNXRUNTIME,
             LangRec.JAPAN,
-            ModelType.MOBILE,
-            OCRVersion.PPOCRV4,
+            ModelType.SMALL,
+            OCRVersion.PPOCRV6,
             "No",
             ["use_det", "use_cls", "use_rec"],
         ],
@@ -191,9 +192,9 @@ def create_examples() -> List[List[Union[str, float]]]:
             1.6,
             2000,
             EngineType.ONNXRUNTIME,
-            LangDet.MULTI,
+            LangDet.CH,
             ModelType.MOBILE,
-            OCRVersion.PPOCRV4,
+            OCRVersion.PPOCRV5,
             EngineType.ONNXRUNTIME,
             LangCls.CH,
             ModelType.MOBILE,
@@ -201,7 +202,7 @@ def create_examples() -> List[List[Union[str, float]]]:
             EngineType.ONNXRUNTIME,
             LangRec.KOREAN,
             ModelType.MOBILE,
-            OCRVersion.PPOCRV4,
+            OCRVersion.PPOCRV5,
             "No",
             ["use_det", "use_cls", "use_rec"],
         ],
@@ -457,7 +458,7 @@ with gr.Blocks(
         label="OCR Txts",
         headers=["Index", "Txt", "Score"],
         datatype=["number", "str", "number"],
-        show_copy_button=True,
+        buttons=["copy"],
     )
 
     ocr_inputs = [