Update app.py
Browse files
app.py
CHANGED
@@ -214,14 +214,14 @@ def translate_mt5_myen(source, translator, tokenizer):
|
|
214 |
def call_model_mt5(source, direction_mt5):
|
215 |
if direction_mt5 == "English to Myanmar":
|
216 |
translator = ctranslate2.Translator(mt5_ct_model_path)
|
217 |
-
tokenizer = transformers.AutoTokenizer.from_pretrained(
|
218 |
if source == "" :
|
219 |
gr.Warning("Please Enter English Text")
|
220 |
else:
|
221 |
translation = translate_mt5_enmy(source, translator, tokenizer)
|
222 |
elif direction_mt5 == "Myanmar to English":
|
223 |
translator = ctranslate2.Translator(mt5_ct_model_path)
|
224 |
-
tokenizer = transformers.AutoTokenizer.from_pretrained(
|
225 |
if source == "" :
|
226 |
gr.Warning("Please Enter Myanmar Text")
|
227 |
else:
|
|
|
214 |
def call_model_mt5(source, direction_mt5):
|
215 |
if direction_mt5 == "English to Myanmar":
|
216 |
translator = ctranslate2.Translator(mt5_ct_model_path)
|
217 |
+
tokenizer = transformers.AutoTokenizer.from_pretrained("t5-base")
|
218 |
if source == "" :
|
219 |
gr.Warning("Please Enter English Text")
|
220 |
else:
|
221 |
translation = translate_mt5_enmy(source, translator, tokenizer)
|
222 |
elif direction_mt5 == "Myanmar to English":
|
223 |
translator = ctranslate2.Translator(mt5_ct_model_path)
|
224 |
+
tokenizer = transformers.AutoTokenizer.from_pretrained("t5-base")
|
225 |
if source == "" :
|
226 |
gr.Warning("Please Enter Myanmar Text")
|
227 |
else:
|