浏览代码

fix(model): make language parameter optional in get_model

myhloli 1 年之前
父节点
当前提交
f9841a0c2b
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      magic_pdf/model/doc_analyze_by_custom_model.py

+ 1 - 1
magic_pdf/model/doc_analyze_by_custom_model.py

@@ -57,7 +57,7 @@ class ModelSingleton:
             cls._instance = super().__new__(cls)
         return cls._instance
 
-    def get_model(self, ocr: bool, show_log: bool, lang):
+    def get_model(self, ocr: bool, show_log: bool, lang=None):
         key = (ocr, show_log, lang)
         if key not in self._models:
             self._models[key] = custom_model_init(ocr=ocr, show_log=show_log, lang=lang)