Update app.py

app.py

@@ -7,7 +7,7 @@ import httpx
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 # Language options and mappings
-options = ["German", "Romanian", "English", "French", "Spanish", "Italian"]
+options = ["German", "Romanian", "English", "French", "Spanish", "Italian", "Hindi"]
 favourite_langs = {"German": "de", "Romanian": "ro", "English": "en", "-----": "-----"}
 df = pl.read_parquet("isolanguages.parquet")
 non_empty_isos = df.slice(1).filter(pl.col("ISO639-1") != "").rows()
@@ -20,7 +20,7 @@ iso1_to_name = {iso[1]: iso[0] for iso in non_empty_isos} # {'ro': 'Romanian', '
 langs = {iso[0]: iso[1] for iso in non_empty_isos} # {'Romanian': 'ro', 'German': 'de'}
 
 
-models = ["Helsinki-NLP", "Argos", "Google", "t5-base", "t5-small", "t5-large", "Unbabel/Tower-Plus-2B", 
+models = ["Helsinki-NLP", "Argos", "Google", "t5-base", "t5-small", "t5-large", "HPLT", "Unbabel/Tower-Plus-2B", 
           "Unbabel/TowerInstruct-Mistral-7B-v0.2", "winninghealth/WiNGPT-Babel-2", "towing/viachat-t5-large-v3.0", "towing/viachat-t5-large-v2.0"]
 allmodels = ["Helsinki-NLP",
           "Helsinki-NLP/opus-mt-tc-bible-big-mul-mul", "Helsinki-NLP/opus-mt-tc-bible-big-mul-deu_eng_nld",
@@ -52,6 +52,11 @@ class Translators:
         response = httpx.get(url)
         return response.json()[0][0][0]
 
+    def hplt(self):
+    model = f'HPLT/translate-{self.sl}-{self.tl}-v1.0-hplt'
+    pipe = pipeline("translation", model=model)
+    return translation[0]['translation_text']
+    
     @classmethod
     def download_argos_model(cls, from_code, to_code):
         import argostranslate.package
@@ -418,6 +423,9 @@ def translate_text(model_name: str, s_language: str, t_language: str, input_text
         
         elif model_name == "Helsinki-NLP":
             translated_text, message_text = Translators(model_name, sl, tl, input_text).HelsinkiNLP()
+
+        elif model_name == "HPLT":
+            translated_text, message_text = Translators(model_name, sl, tl, input_text).hplt()
         
         elif model_name == 'Argos':
             translated_text = Translators(model_name, sl, tl, input_text).argos()