TiberiuCristianLeon commited on
Commit
135432f
·
verified ·
1 Parent(s): 28cef2e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -1
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}'