TiberiuCristianLeon commited on
Commit
0b61062
·
verified ·
1 Parent(s): 4894c8f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -20,10 +20,12 @@ def model_to_cuda(model):
20
  else:
21
  print("CUDA not available! Using CPU.")
22
  return model
 
23
  @spaces.GPU
24
  def translate_text(input_text, sselected_language, tselected_language, model_name):
25
  sl = all_langs[sselected_language]
26
  tl = all_langs[tselected_language]
 
27
  if model_name == "Helsinki-NLP":
28
  try:
29
  model_name = f"Helsinki-NLP/opus-mt-{sl}-{tl}"
 
20
  else:
21
  print("CUDA not available! Using CPU.")
22
  return model
23
+
24
  @spaces.GPU
25
  def translate_text(input_text, sselected_language, tselected_language, model_name):
26
  sl = all_langs[sselected_language]
27
  tl = all_langs[tselected_language]
28
+ message_text = f'Translated from {sselected_language} to {tselected_language} with {model_name}'
29
  if model_name == "Helsinki-NLP":
30
  try:
31
  model_name = f"Helsinki-NLP/opus-mt-{sl}-{tl}"