Update app.py
Browse files
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 |
-
|
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
|