mbarnig commited on
Commit
a0edb0d
β€’
1 Parent(s): 273c538

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -45,9 +45,9 @@ def customization(myModel, direc):
45
  else:
46
  print("Please select a Translation Direction !")
47
  elif myModel == "T5/mt5":
48
- # translator = pipeline("translation", model=REPO_ID_T5MT5)
49
- tokenizer = AutoTokenizer.from_pretrained("mbarnig/T5-mt5-taboeta-en-lb")
50
- model = AutoModelForSeq2SeqLM.from_pretrained("mbarnig/T5-mt5-taboeta-en-lb")
51
  else:
52
  print("Please select a Translation Model !")
53
  return model, tokenizer
 
45
  else:
46
  print("Please select a Translation Direction !")
47
  elif myModel == "T5/mt5":
48
+ translator = pipeline("translation", model=REPO_ID_T5MT5)
49
+ # tokenizer = AutoTokenizer.from_pretrained("mbarnig/T5-mt5-taboeta-en-lb")
50
+ # model = AutoModelForSeq2SeqLM.from_pretrained("mbarnig/T5-mt5-taboeta-en-lb")
51
  else:
52
  print("Please select a Translation Model !")
53
  return model, tokenizer