NgalNgal commited on
Commit
90ad9c0
1 Parent(s): d7577fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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(mt5_sp_model_path)
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(mt5_sp_model_path)
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: