Update app.py

app.py

@@ -342,6 +342,21 @@ class Translators:
         translated_text = '\n'.join(split_translated_text[:input_text.count('\n')+1])
         return translated_text
 
+def bergamot(model_name: str = 'deen', sl: str = 'de', tl: str = 'en', input_text: str):
+    try:
+        import bergamot
+        # input_text = [input_text] if isinstance(input_text, str) else input_text           
+        config = bergamot.ServiceConfig(numWorkers=4)
+        service = bergamot.Service(config)
+        model = service.modelFromConfigPath(f"./{model_name}/bergamot.config.yml")
+        options = bergamot.ResponseOptions(alignment=False, qualityScores=False, HTML=False)
+        rawresponse = service.translate(model, bergamot.VectorString(input_text), options)
+        translated_text: str = next(iter(rawresponse)).target.text
+        message_text = f"Translated from {sl} to {tl} with {model_name}."
+    except Exception as error:
+        response = error
+    return translated_text, message_text
+
 @spaces.GPU
 def translate_text(input_text: str, s_language: str, t_language: str, model_name: str) -> tuple[str, str]:
     """
@@ -424,6 +439,9 @@ def translate_text(input_text: str, s_language: str, t_language: str, model_name
 
         elif model_name == "winninghealth/WiNGPT-Babel-2":      
             translated_text = Translators(model_name, s_language, t_language, input_text).wingpt()
+
+        elif model_name == "Bergamot":
+            translated_text, message_text = bergamot(model_name, s_language, t_language, input_text)
             
     except Exception as error:
         translated_text = error