Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -104,7 +104,19 @@ def nllb(model_name, sl, tl, input_text):
|
|
| 104 |
return translated_text[0]['translation_text']
|
| 105 |
|
| 106 |
@spaces.GPU
|
| 107 |
-
def translate_text(input_text, sselected_language, tselected_language, model_name):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 108 |
sl = all_langs[sselected_language]
|
| 109 |
tl = all_langs[tselected_language]
|
| 110 |
message_text = f'Translated from {sselected_language} to {tselected_language} with {model_name}'
|
|
|
|
| 104 |
return translated_text[0]['translation_text']
|
| 105 |
|
| 106 |
@spaces.GPU
|
| 107 |
+
def translate_text(input_text: str, sselected_language: str, tselected_language: str, model_name: str) -> str:
|
| 108 |
+
"""
|
| 109 |
+
Translates the input text from the source language to the target language.
|
| 110 |
+
|
| 111 |
+
Args:
|
| 112 |
+
input_text (str): The source text to be translated
|
| 113 |
+
sselected_language (str): The source language of the input text
|
| 114 |
+
tselected_language (str): The target language in which the input text is translated
|
| 115 |
+
model (str): The selected translation model
|
| 116 |
+
|
| 117 |
+
Returns:
|
| 118 |
+
str: The input text translated to the selected target language
|
| 119 |
+
"""
|
| 120 |
sl = all_langs[sselected_language]
|
| 121 |
tl = all_langs[tselected_language]
|
| 122 |
message_text = f'Translated from {sselected_language} to {tselected_language} with {model_name}'
|