Update app.py

app.py

@@ -63,24 +63,23 @@ class Translators:
         from quickmt.hub import hf_download
         from pathlib import Path
         model_name = f"quickmt-{self.sl}-{self.tl}"
-        model_path = Path("/code/models") / model_name
-        from quickmt.hub import hf_list
-        choices = [i.split("/quickmt-")[1] for i in hf_list()]
-        if f"{self.sl}-{self.tl}" not in choices:
+        model_path = Path("/quickmt/models") / model_name
+        # from quickmt.hub import hf_list
+        # quickmt_models = [i.split("/quickmt-")[1] for i in hf_list()]
+        # quickmt_models.sort()
+        # print(quickmt_models)
+        quickmt_models = ['ar-en', 'bn-en', 'cs-en', 'da-en', 'de-en', 'el-en', 'en-ar', 'en-bn', 'en-cs', 'en-de', 'en-el', 'en-es',
+                  'en-fa', 'en-fr', 'en-he', 'en-hi', 'en-hu', 'en-id', 'en-it', 'en-ja', 'en-ko', 'en-lv', 'en-pl', 'en-pt',
+                  'en-ro', 'en-ru', 'en-th', 'en-tr', 'en-ur', 'en-vi', 'en-zh', 'es-en', 'fa-en', 'fr-en', 'he-en', 'hi-en',
+                  'hu-en', 'id-en', 'it-en', 'ja-en', 'ko-en', 'lv-en', 'pl-en', 'pt-en', 'ro-en', 'ru-en', 'th-en', 'tr-en', 'ur-en', 'vi-en', 'zh-en']
+        if f"{self.sl}-{self.tl}" not in quickmt_models:
             translation = f'Model {model_name} from {iso1_to_name[self.sl]} to {iso1_to_name[self.tl]} not available!'
-            message = f"Available models: {', '.join(choices.sort())}"
+            message = f"Available models: {', '.join(quickmt_models)}"
             return translation, message
-        models = ['zh-en', 'en-zh', 'fr-en', 'en-fr', 'de-en', 'en-de', 'ko-en',
-                  'en-ko', 'en-ar', 'ar-en', 'ja-en', 'en-ja', 'it-en', 'en-it',
-                  'hi-en', 'en-hi', 'en-es', 'es-en', 'ru-en', 'en-ru', 'pt-en',
-                  'en-pt', 'bn-en', 'en-bn', 'id-en', 'en-id', 'fa-en', 'en-fa',
-                  'vi-en', 'en-vi', 'tr-en', 'en-tr', 'ur-en', 'en-ur', 'en-hu',
-                  'hu-en', 'th-en', 'en-th', 'el-en', 'en-el', 'pl-en', 'en-pl',
-                  'en-lv', 'lv-en', 'cs-en', 'ro-en', 'en-ro', 'en-cs', 'he-en', 'en-he', 'da-en']
         if not model_path.exists():
             hf_download(
             model_name = f"quickmt/{model_name}",
-            output_dir=Path("/code/models") / model_name,
+            output_dir=Path("/quickmt/models") / model_name,
         )
         # 'auto' auto-detects GPU, set to "cpu" to force CPU inference
         device = 'gpu' if torch.cuda.is_available() else 'cpu'