Update app.py

app.py

@@ -65,6 +65,7 @@ class Translators:
         self.model_name = model_name
         self.sl, self.tl = sl, tl
         self.input_text = input_text
+        self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         
     def google(self):  
         url = os.environ['GCLIENT'] + f'sl={self.sl}&tl={self.tl}&q={self.input_text}'
@@ -98,8 +99,7 @@ class Translators:
 
     def HelsinkiNLP_mulmul(self):
         try:
-            model_name = "Helsinki-NLP/opus-mt-tc-bible-big-mul-mul"
-            pipe = pipeline("translation", model=model_name)                
+            pipe = pipeline("translation", model=self.model_name, device=self.device)                
             iso1_dict = {iso[1]: (iso[0], iso[2], iso[3]) for iso in non_empty_isos}
             iso3tl = iso1_dict.get(self.tl)[2] # 'deu', 'ron', 'eng', 'fra'
             translation = pipe(f'>>{iso3tl}<< {self.input_text}')
@@ -110,17 +110,18 @@ class Translators:
     def HelsinkiNLP(self):
         try: # Standard bilingual model
             model_name = f"Helsinki-NLP/opus-mt-{self.sl}-{self.tl}"
-            pipe = pipeline("translation", model=model_name, device=-1)
+            pipe = pipeline("translation", model=model_name, device=self.device)
             translation = pipe(self.input_text)
             return translation[0]['translation_text'], f'Translated from {self.sl} to {self.tl} with {model_name}.'
         except EnvironmentError:
             try: # Tatoeba models
                 model_name = f"Helsinki-NLP/opus-tatoeba-{self.sl}-{self.tl}"
-                pipe = pipeline("translation", model=model_name, device=-1)
+                pipe = pipeline("translation", model=model_name, device=self.device)
                 translation = pipe(self.input_text)
                 return translation[0]['translation_text'], f'Translated from {self.sl} to {self.tl} with {model_name}.'
             except EnvironmentError as error:
-                return self.HelsinkiNLP_mulmul() # Last resort: multi to multi
+                self.model_name = "Helsinki-NLP/opus-mt-tc-bible-big-mul-mul" # Last resort: try multi to multi
+                return self.HelsinkiNLP_mulmul()
         except KeyError as error:
             return f"Error: Translation direction {self.sl} to {self.tl} is not supported by Helsinki Translation Models", error
         
@@ -363,7 +364,7 @@ def translate_text(input_text: str, s_language: str, t_language: str, model_name
         if model_name == "Helsinki-NLP/opus-mt-tc-bible-big-mul-mul":
             translated_text, message_text = Translators(model_name, sl, tl, input_text).HelsinkiNLP_mulmul()
         
-        elif model_name.startswith("Helsinki-NLP"):
+        elif model_name == "Helsinki-NLP":
             translated_text, message_text = Translators(model_name, sl, tl, input_text).HelsinkiNLP()
         
         elif model_name == 'Argos':