Update app.py

app.py

@@ -358,20 +358,20 @@ 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 = 'Hallo, mein Freund'):
-    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 Bergamot {model_name}."
-    except Exception as error:
-        response = error
-    return translated_text, message_text
+    def bergamot(model_name: str = 'deen', sl: str = 'de', tl: str = 'en', input_text: str = 'Hallo, mein Freund'):
+        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 Bergamot {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]:
@@ -460,7 +460,7 @@ def translate_text(input_text: str, s_language: str, t_language: str, model_name
             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)
+            translated_text, message_text = Translators(model_name, s_language, t_language, input_text).bergamot()
             
     except Exception as error:
         translated_text = error
@@ -503,7 +503,7 @@ def create_interface():
         message_text = gr.Textbox(label="Messages:", placeholder="Display field for status and error messages", interactive=False,
                                   value=f'Default translation settings: from {s_language.value} to {t_language.value} with {model_name.value}.')
         allmodels = gr.HTML(label="Model links:", value=', '.join([f'<a href="https://huggingface.co/{model}">{model}</a>' for model in models]))
-        model_info = gr.Markdown(label="Model info:", value='### [Helsinki-NLP](https://huggingface.co/Helsinki-NLP "Helsinki-NLP"', show_copy_button=True)
+        model_info = gr.Markdown(label="Model info:", value='### [Helsinki-NLP](https://huggingface.co/Helsinki-NLP "Helsinki-NLP")', show_copy_button=True)
         model_name.change(fn=get_info, inputs=[model_name, s_language, t_language], outputs=model_info, api_name=False, show_api=False)
         
         translate_btn.click(