mirror of
https://github.com/RapidAI/RapidOCR.git
synced 2026-11-10 21:10:55 +08:00
1 parent
402aadb121
commit
58bf4f81cc
1 file changed
+6
-1
@@ -1,4 +1,4 @@
|
||||
# -*- encoding: utf-8 -*-
|
||||
# -*- encoding: utf-8 -*-
|
||||
# @Author: SWHL
|
||||
# @Contact: liekkaskono@163.com
|
||||
import abc
|
||||
@@ -113,6 +113,11 @@ class InferSession(abc.ABC):
|
||||
cls.model_info, f"{engine_type}.{ocr_version}.{task_type}"
|
||||
)
|
||||
|
||||
if not model_dict:
|
||||
raise ValueError(
|
||||
f"Unsupported configuration: {engine_type}.{ocr_version}.{task_type}.{model_type}"
|
||||
)
|
||||
|
||||
# 优先查找 server 模型
|
||||
if model_type == ModelType.SERVER.value:
|
||||
for k in model_dict:
|
||||
|
||||
Reference in new issue
Block a user