Update app.py
Browse files
app.py
CHANGED
|
@@ -94,17 +94,17 @@ class Translators:
|
|
| 94 |
available_languages = ['ar', 'bn', 'cs', 'da', 'de', 'el', 'en', 'es', 'fa', 'fr', 'he', 'hi', 'hu',
|
| 95 |
'id', 'it', 'ja', 'ko', 'lv', 'pl', 'pt', 'ro', 'ru', 'th', 'tr', 'ur', 'vi', 'zh']
|
| 96 |
if f"{self.sl}-{self.tl}" in quickmt_models:
|
| 97 |
-
model_path = quickmtdownload(model_name)
|
| 98 |
-
translation = quickmttranslate(model_path, self.input_text)
|
| 99 |
message = f'Translated from {iso1_to_name[self.sl]} to {iso1_to_name[self.tl]} with {model_name}.'
|
| 100 |
# Pivot language English
|
| 101 |
elif self.sl in available_languages and self.tl in available_languages:
|
| 102 |
model_name = f"quickmt-{self.sl}-en"
|
| 103 |
-
model_path = quickmtdownload(model_name)
|
| 104 |
-
entranslation = quickmttranslate(model_path, self.input_text)
|
| 105 |
model_name = f"quickmt-en-{self.tl}"
|
| 106 |
-
model_path = quickmtdownload(model_name)
|
| 107 |
-
translation = quickmttranslate(model_path, entranslation)
|
| 108 |
message = f'Translated from {iso1_to_name[self.sl]} to {iso1_to_name[self.tl]} with pivot language English.'
|
| 109 |
else:
|
| 110 |
translation = f'Model {model_name} from {iso1_to_name[self.sl]} to {iso1_to_name[self.tl]} not available!'
|
|
|
|
| 94 |
available_languages = ['ar', 'bn', 'cs', 'da', 'de', 'el', 'en', 'es', 'fa', 'fr', 'he', 'hi', 'hu',
|
| 95 |
'id', 'it', 'ja', 'ko', 'lv', 'pl', 'pt', 'ro', 'ru', 'th', 'tr', 'ur', 'vi', 'zh']
|
| 96 |
if f"{self.sl}-{self.tl}" in quickmt_models:
|
| 97 |
+
model_path = Translators.quickmtdownload(model_name)
|
| 98 |
+
translation = Translators.quickmttranslate(model_path, self.input_text)
|
| 99 |
message = f'Translated from {iso1_to_name[self.sl]} to {iso1_to_name[self.tl]} with {model_name}.'
|
| 100 |
# Pivot language English
|
| 101 |
elif self.sl in available_languages and self.tl in available_languages:
|
| 102 |
model_name = f"quickmt-{self.sl}-en"
|
| 103 |
+
model_path = Translators.quickmtdownload(model_name)
|
| 104 |
+
entranslation = Translators.quickmttranslate(model_path, self.input_text)
|
| 105 |
model_name = f"quickmt-en-{self.tl}"
|
| 106 |
+
model_path = Translators.quickmtdownload(model_name)
|
| 107 |
+
translation = Translators.quickmttranslate(model_path, entranslation)
|
| 108 |
message = f'Translated from {iso1_to_name[self.sl]} to {iso1_to_name[self.tl]} with pivot language English.'
|
| 109 |
else:
|
| 110 |
translation = f'Model {model_name} from {iso1_to_name[self.sl]} to {iso1_to_name[self.tl]} not available!'
|