TiberiuCristianLeon commited on
Commit
784f38c
·
verified ·
1 Parent(s): 966bf4d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
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!'