Add spanish

This commit is contained in:
WeberJulian 2023-11-15 14:35:30 +01:00
parent 2d9525a669
commit d64cafccd2
1 changed files with 4 additions and 0 deletions

View File

@ -16,6 +16,7 @@ from spacy.lang.en import English
from spacy.lang.zh import Chinese from spacy.lang.zh import Chinese
from spacy.lang.ja import Japanese from spacy.lang.ja import Japanese
from spacy.lang.ar import Arabic from spacy.lang.ar import Arabic
from spacy.lang.es import Spanish
def get_spacy_lang(lang): def get_spacy_lang(lang):
@ -25,7 +26,10 @@ def get_spacy_lang(lang):
return Japanese() return Japanese()
elif lang == "ar": elif lang == "ar":
return Arabic() return Arabic()
elif lang == "es":
return Spanish()
else: else:
# For most languages, Enlish does the job
return English() return English()
def split_sentence(text, lang, text_split_length=250): def split_sentence(text, lang, text_split_length=250):