Update app.py

app.py

@@ -66,8 +66,8 @@ class Translators:
         translator = Translator(str(model_path), device = device)       
         # translation = Translator(f"./quickmt-{self.sl}-{self.tl}/", device="auto", inter_threads=2)       
         # set beam size to 1 for faster speed (but lower quality)
-        translation = translator(input_text, beam_size=5)
-        print(model_path, input_text, translation)
+        translation = translator(input_text, beam_size=5, compute_type="float16")
+        # print(model_path, input_text, translation)
         return translation
 
     @staticmethod
@@ -96,6 +96,7 @@ class Translators:
         # available_languages.sort()
         available_languages = ['ar', 'bn', 'cs', 'da', 'de', 'el', 'en', 'es', 'fa', 'fr', 'he', 'hi', 'hu',
                                'id', 'it', 'ja', 'ko', 'lv', 'pl', 'pt', 'ro', 'ru', 'th', 'tr', 'ur', 'vi', 'zh']
+        # Direct translation model
         if f"{self.sl}-{self.tl}" in quickmt_models:
             model_path = Translators.quickmtdownload(model_name)
             translation = Translators.quickmttranslate(model_path, self.input_text)
@@ -109,7 +110,7 @@ class Translators:
             model_path = Translators.quickmtdownload(model_name)
             translation = Translators.quickmttranslate(model_path, entranslation)
             message = f'Translated from {iso1_to_name[self.sl]} to {iso1_to_name[self.tl]} with pivot language English.'
-        else:                   
+        else:
             translation = f'Model {model_name} from {iso1_to_name[self.sl]} to {iso1_to_name[self.tl]} not available!'
             message = f"Available models: {', '.join(quickmt_models)}"
         return translation, message