Browse Source

fix(model): make language parameter optional in get_model

myhloli 1 year ago
parent
commit
f9841a0c2b
1 changed files with 1 additions and 1 deletions
  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)