Move folders
This commit is contained in:
parent
ba2e252759
commit
3cf28379e8
89 changed files with 47 additions and 31 deletions
16
src/great_ai/utilities/language/predict_language.py
Normal file
16
src/great_ai/utilities/language/predict_language.py
Normal file
|
|
@ -0,0 +1,16 @@
|
|||
from typing import Optional
|
||||
|
||||
from langcodes import Language
|
||||
from langdetect import LangDetectException, detect
|
||||
|
||||
|
||||
def predict_language(text: Optional[str]) -> str:
|
||||
if not text:
|
||||
return Language.make().to_tag()
|
||||
|
||||
try:
|
||||
language_code = detect(text)
|
||||
except LangDetectException:
|
||||
return Language.make().to_tag()
|
||||
|
||||
return Language.get(language_code).to_tag()
|
||||
Loading…
Add table
Add a link
Reference in a new issue