hplt zh to zh_hunt code

app.py

@@ -65,8 +65,8 @@ class Translators:
     def mitre(self):
         from transformers import AutoModel, AutoTokenizer
         tokenizer = AutoTokenizer.from_pretrained(self.model_name, trust_remote_code=True, use_fast=False)
-        # model = AutoModel.from_pretrained(self.model_name, trust_remote_code=True, use_fast=False).to(self.device)
-        model = AutoModel.from_pretrained(self.model_name, trust_remote_code=True)
+        model = AutoModel.from_pretrained(self.model_name, trust_remote_code=True, use_fast=False).to(self.device)
+        # model = AutoModel.from_pretrained(self.model_name, trust_remote_code=True)
         # model.half() # recommended for GPU
         model.eval()
         # Translating from one or several sentences to a sole language
@@ -88,6 +88,9 @@ class Translators:
                   'en-ga', 'en-gl', 'en-hi', 'en-hr', 'en-is', 'en-mt', 'en-nn', 'en-sq', 'en-sw',
                   'en-zh_hant', 'et-en', 'eu-en', 'fi-en', 'ga-en', 'gl-en', 'hi-en', 'hr-en',
                   'is-en', 'mt-en', 'nn-en', 'sq-en', 'sw-en', 'zh_hant-en']
+        lang_map = {"zh": "zh_hant"}
+        self.sl = lang_map.get(self.sl, self.sl)
+        self.tl = lang_map.get(self.tl, self.tl)
         if opus:
             hplt_model = f'HPLT/translate-{self.sl}-{self.tl}-v1.0-hplt_opus' # HPLT/translate-en-hr-v1.0-hplt_opus
         else: