BounharAbdelaziz commited on
Commit
2a21734
·
verified ·
1 Parent(s): 4e0cddf

Ultra+Supreme Remove src_lang and tgt_lang in load

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -32,13 +32,13 @@ def translate_nano_large(text, model_path):
32
  # Translation function for Ultra and Supreme models
33
  def translate_ultra_supreme(text, model_path):
34
  model = AutoModelForSeq2SeqLM.from_pretrained(model_path, token=TOKEN)
35
- tokenizer = AutoTokenizer.from_pretrained(model_path, src_lang="eng_Latn", tgt_lang="ary_Arab", token=TOKEN)
36
  translator = pipeline(
37
  "translation",
38
  model=model,
39
  tokenizer=tokenizer,
40
  max_length=512,
41
- src_lang="eng_Latn",
42
  tgt_lang="ary_Arab"
43
  )
44
  translation = translator(text)[0]['translation_text']
 
32
  # Translation function for Ultra and Supreme models
33
  def translate_ultra_supreme(text, model_path):
34
  model = AutoModelForSeq2SeqLM.from_pretrained(model_path, token=TOKEN)
35
+ tokenizer = AutoTokenizer.from_pretrained(model_path, token=TOKEN) # Remove src_lang and tgt_lang
36
  translator = pipeline(
37
  "translation",
38
  model=model,
39
  tokenizer=tokenizer,
40
  max_length=512,
41
+ src_lang="eng_Latn", # Keep src_lang and tgt_lang in the pipeline
42
  tgt_lang="ary_Arab"
43
  )
44
  translation = translator(text)[0]['translation_text']