Snigdhapaul2003 commited on
Commit
6dc3987
·
verified ·
1 Parent(s): 8d57e67

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -43,9 +43,10 @@ languages = {
43
  'Welsh': 'cym_Latn'
44
  }
45
 
 
 
 
46
  def translate_article(article, language):
47
- tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
48
- model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
49
  inputs = tokenizer(article, return_tensors="pt")
50
  lang_code = languages[language]
51
  translated_tokens = model.generate(**inputs, forced_bos_token_id=tokenizer.lang_code_to_id[lang_code])
 
43
  'Welsh': 'cym_Latn'
44
  }
45
 
46
+ tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
47
+ model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
48
+
49
  def translate_article(article, language):
 
 
50
  inputs = tokenizer(article, return_tensors="pt")
51
  lang_code = languages[language]
52
  translated_tokens = model.generate(**inputs, forced_bos_token_id=tokenizer.lang_code_to_id[lang_code])