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,
@@ -23,6 +24,7 @@ def get_ocr_result(
     text_score,
     box_thresh,
     unclip_ratio,
+    max_side_len,
     det_engine,
     lang_det,
     det_model_type,
@@ -46,6 +48,7 @@ def get_ocr_result(
 
     ocr_engine = RapidOCR(
         params={
+            "Global.max_side_len": max_side_len,
             "Det.engine_type": EngineType(det_engine),
             "Det.lang_type": LangDet(lang_det),
             "Det.model_type": ModelType(det_model_type),
@@ -95,6 +98,7 @@ def create_examples() -> List[List[Union[str, float]]]:
             0.5,
             0.5,
             1.6,
+            2000,
             EngineType.ONNXRUNTIME,
             LangDet.CH,
             ModelType.MOBILE,
@@ -115,6 +119,7 @@ def create_examples() -> List[List[Union[str, float]]]:
             0.5,
             0.5,
             1.6,
+            2000,
             EngineType.ONNXRUNTIME,
             LangDet.CH,
             ModelType.MOBILE,
@@ -135,6 +140,7 @@ def create_examples() -> List[List[Union[str, float]]]:
             0.5,
             0.5,
             1.6,
+            2000,
             EngineType.ONNXRUNTIME,
             LangDet.CH,
             ModelType.MOBILE,
@@ -155,6 +161,7 @@ def create_examples() -> List[List[Union[str, float]]]:
             0.5,
             0.5,
             1.6,
+            2000,
             EngineType.ONNXRUNTIME,
             LangDet.MULTI,
             ModelType.MOBILE,
@@ -175,6 +182,7 @@ def create_examples() -> List[List[Union[str, float]]]:
             0.5,
             0.5,
             1.6,
+            2000,
             EngineType.ONNXRUNTIME,
             LangDet.MULTI,
             ModelType.MOBILE,
@@ -199,6 +207,7 @@ def export_yaml(
     text_score,
     box_thresh,
     unclip_ratio,
+    max_side_len,
     det_engine,
     lang_det,
     det_model_type,
@@ -219,6 +228,7 @@ def export_yaml(
 
     params = {
         "Global": {
+            "max_side_len": max_side_len,
             "use_det": "use_det" in use_module,
             "use_cls": "use_cls" in use_module,
             "use_rec": "use_rec" in use_module,
@@ -311,6 +321,13 @@ with gr.Blocks(
                 step=0.1,
                 info="控制文本检测框的大小，值越大，检测框整体越大。在出现框截断文字的情况，调大该值。取值范围：[1.5, 2.0]，默认值为1.6",
             )
+            max_side_len = gr.Number(
+                value=2000,
+                label="max_side_len",
+                info="如果输入图像的最大边大于`max_side_len`，则会按宽高比，将最大边缩放到`max_side_len`。默认为2000px",
+                interactive=True,
+                minimum=20,
+            )
 
         with gr.Row():
             with gr.Row():
@@ -429,6 +446,7 @@ with gr.Blocks(
         text_score,
         box_thresh,
         unclip_ratio,
+        max_side_len,
         det_engine,
         lang_det,
         det_model_type,