Zeimoto commited on
Commit
3e48293
·
1 Parent(s): e25d134

added print in model name

Browse files
Files changed (1) hide show
  1. translation.py +1 -0
translation.py CHANGED
@@ -6,6 +6,7 @@ def get_model_name(languageCode: str) -> str:
6
 
7
  def init_translation_model():
8
  model_name = get_model_name("pt")
 
9
  model = MarianMTModel.from_pretrained(model_name)
10
  tokenizer = MarianTokenizer.from_pretrained(model_name)
11
  return model, tokenizer
 
6
 
7
  def init_translation_model():
8
  model_name = get_model_name("pt")
9
+ print(model_name)
10
  model = MarianMTModel.from_pretrained(model_name)
11
  tokenizer = MarianTokenizer.from_pretrained(model_name)
12
  return model, tokenizer