mbarnig commited on
Commit
0a3a588
β€’
1 Parent(s): f57d6cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ my_output = gr.Textbox(lines=5, label="Translation")
36
 
37
  def iwwersetz(source_text, model, direc):
38
  if model == "NLLB":
39
- translator = pipeline("translation", model=model_checkpoint)
40
  if direc == "en -> lb":
41
  # translation = translator("en", "lb", source_text)
42
  translation = source_text
 
36
 
37
  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("en", "lb", source_text)
42
  translation = source_text