Hugging Face's logo

Spaces:
mbarnig
/
translation-lb-en-with-3-models
Running

App Files Community
mbarnig commited on
Commit
2164c20
•
1 Parent(s): 0e7c921

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -38,10 +38,10 @@ def iwwersetz(source_text, model, direc):
38
  if model == "NLLB":
39
  translator = pipeline("translation", model=model_checkpoint_nllb)
40
  if direc == "en -> lb":
41
- translation = translator(source_text, src_lang="eng_Latn", tgt_lan="ltz_Latn")
42
  # translation = source_text
43
  else:
44
- translation = translator(source_text, src_lan="ltz_Latn", tgt_lang="eng_Latn")
45
  # translation = source_text
46
  elif model == "MarianNMT":
47
  if direc == "en -> lb":
 
38
  if model == "NLLB":
39
  translator = pipeline("translation", model=model_checkpoint_nllb)
40
  if direc == "en -> lb":
41
+ translation = translator(source_text, src_lang="eng_Latn", tgt_lang="ltz_Latn")
42
  # translation = source_text
43
  else:
44
+ translation = translator(source_text, src_lang="ltz_Latn", tgt_lang="eng_Latn")
45
  # translation = source_text
46
  elif model == "MarianNMT":
47
  if direc == "en -> lb":