don't pass newlines to fasttext. this will fail.

slohmaier · Jul 5, 2022 · e70fe18 · e70fe18
1 parent 97a4984
commit e70fe18
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/addon/globalPlugins/langPredict.py b/addon/globalPlugins/langPredict.py
@@ -96,6 +96,7 @@ def predictLang(langChangeCmd: LangChangeCommand, text: str):
 	defaultLang = synth.availableVoices[synth.voice].language
 	if langChangeCmd is None:
 		langChangeCmd = LangChangeCommand(defaultLang)
+	text = text.replace('\n', ' ').replace('\r', ' ') #fasttext doe not like newlines
 	predictedLang = fastTextModel.predict(text)[0][0][9:] #strip '__label__'
 	log.debug('PREDICTED={0} TEXT={1}'.format(str(predictedLang), text))
 

diff --git a/buildVars.py b/buildVars.py
@@ -25,7 +25,7 @@ def _(arg):
 	# Translators: Long description to be shown for this add-on on add-on information from add-ons manager
 	"addon_description": _("""Predict the language of every spoken text with an ai model and change the voice accordingly, if available."""),
 	# version
-	"addon_version": "0.0.1",
+	"addon_version": "0.0.2",
 	# Author(s)
 	"addon_author": "Stefan Lohmaier <stefan@slohmaier.de>",
 	# URL for the add-on documentation support