Skip to content

GitLab

Unverified Commit e5b74ae7 authored by Xiaomeng Zhao's avatar Xiaomeng Zhao Committed by GitHub
Browse files

Merge pull request #2173 from myhloli/dev 

fix(dataset): correct variable for language detection
parents 1db6f89d 814bd4ea
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
magic_pdf/data/dataset.py
View file @ e5b74ae7
...@@ -249,7 +249,7 @@ class ImageDataset(Dataset): ...@@ -249,7 +249,7 @@ class ImageDataset(Dataset):
elif lang == 'auto': elif lang == 'auto':
from magic_pdf.model.sub_modules.language_detection.utils import \ from magic_pdf.model.sub_modules.language_detection.utils import \
auto_detect_lang 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}') logger.info(f'lang: {lang}, detect_lang: {self._lang}')
else: else:
self._lang = lang self._lang = lang
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment