فهرست منبع

Merge pull request #2173 from myhloli/dev

fix(dataset): correct variable for language detection
Xiaomeng Zhao 7 ماه پیش
والد
کامیت
e5b74ae724
1فایلهای تغییر یافته به همراه1 افزوده شده و 1 حذف شده
  1. 1 1
      magic_pdf/data/dataset.py

+ 1 - 1
magic_pdf/data/dataset.py

@@ -249,7 +249,7 @@ class ImageDataset(Dataset):
         elif lang == 'auto':
             from magic_pdf.model.sub_modules.language_detection.utils import \
                 auto_detect_lang
-            self._lang = auto_detect_lang(bits)
+            self._lang = auto_detect_lang(self._data_bits)
             logger.info(f'lang: {lang}, detect_lang: {self._lang}')
         else:
             self._lang = lang