Merge pull request #2173 from myhloli/dev

fix(dataset): correct variable for language detection

Merge pull request #2173 from myhloli/dev
fix(dataset): correct variable for language detection
e5b74ae7 · Xiaomeng Zhao · GitHub · 1db6f89d · 814bd4ea · e5b74ae7
Unverified Commit e5b74ae7 authored Apr 09, 2025 by Xiaomeng Zhao Committed by GitHub Apr 09, 2025
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

magic_pdf/data/dataset.py magic_pdf/data/dataset.py +1 -1

No files found.
--- a/magic_pdf/data/dataset.py
+++ b/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