use onnx backend for rec models

app.py

@@ -74,7 +74,8 @@ def get_ocr_model(det_model_name, rec_model_name, det_more_configs):
         return MODELS[(det_model_name, rec_model_name, config_str)]
 
     det_model_name, det_model_backend = det_model_name.split('::')
-    rec_model_name, rec_model_backend = rec_model_name.split('::')
+    # rec_model_name, rec_model_backend = rec_model_name.split('::')
+    rec_model_backend = 'onnx'
     model = CnOcr(
         det_model_name=det_model_name,
         det_model_backend=det_model_backend,
@@ -189,7 +190,7 @@ def main():
 
     all_models = list(REC_AVAILABLE_MODELS.all_models())
     all_models.sort()
-    all_models = [f'{m}::{b}' for m, b in all_models]
+    all_models = [f'{m}' for m, b in all_models if b == 'onnx']
 
     title = '开源Python OCR工具：'
     desc = (
@@ -212,49 +213,49 @@ def main():
         [
             'ch_PP-OCRv3_det::onnx',
             True,
-            'number-densenet_lite_136-fc::onnx',
+            'number-densenet_lite_136-fc',
             False,
             'docs/examples/card1-s.jpg',
         ],
         [
             'ch_PP-OCRv3_det::onnx',
             True,
-            'number-densenet_lite_136-fc::onnx',
+            'number-densenet_lite_136-fc',
             False,
             'docs/examples/card2-s.jpg',
         ],
         [
             'ch_PP-OCRv3_det::onnx',
             True,
-            'number-densenet_lite_136-fc::onnx',
+            'number-densenet_lite_136-fc',
             False,
             'docs/examples/cy1-s.jpg',
         ],
         [
             'ch_PP-OCRv3_det::onnx',
             False,
-            'densenet_lite_136-gru::onnx',
+            'densenet_lite_136-gru',
             False,
             'docs/examples/huochepiao.jpeg',
         ],
         [
             'ch_PP-OCRv3_det::onnx',
             False,
-            'densenet_lite_136-gru::onnx',
+            'densenet_lite_136-gru',
             False,
             'docs/examples/1_res.jpg',
         ],
         [
             'db_shufflenet_v2::pytorch',
             False,
-            'en_number_mobile_v2.0::onnx',
+            'en_number_mobile_v2.0',
             False,
             'docs/examples/en_book1.jpeg',
         ],
         [
             'db_shufflenet_v2::pytorch',
             False,
-            'densenet_lite_136-gru::onnx',
+            'densenet_lite_136-gru',
             True,
             'docs/examples/beauty0.jpg',
         ],
@@ -276,7 +277,7 @@ def main():
                 rec_model_name = gr.Dropdown(
                     label='选择识别模型',
                     choices=all_models,
-                    value='densenet_lite_136-fc::onnx',
+                    value='densenet_lite_136-fc',
                 )
 
                 gr.Markdown('### 检测参数')