mirrors/LibreTranslate
1
1
Fork 1
mirror of https://github.com/LibreTranslate/LibreTranslate.git synced 2024-06-10 01:19:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

detect.py: fix bug with language being lowercase

Browse source
This commit is contained in:
Joshix 2023-01-17 20:56:12 +01:00
parent 16718ebd49
commit 53e37fffd9
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
libretranslate/detect.py
View file

@ -41,7 +41,9 @@ class BaseDetector(ABC):
text (string): unicode string.
"""
self.allowed_languages: "frozenset[str]" = frozenset(
self.supported_languages() if allowed_languages is None else allowed_languages
self.supported_languages()
if allowed_languages is None
else [string.upper() for string in allowed_languages]
)
# self.__text = text
self.reliable: bool = True
@ -139,7 +141,7 @@ class Detector(BaseDetector):
confidence_values: "list[tuple[lingua.Language, float]]" = detector.compute_language_confidence_values(text)
return [
Language((language.name.title(), language.iso_code_639_1.name, confidence, 0))
Language((language.name.title(), language.iso_code_639_1.name, confidence, len(text)))
for language, confidence in confidence_values
]