vishwask commited on
Commit
0f79421
·
verified ·
1 Parent(s): e28982d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -35,14 +35,14 @@ translation_tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-lar
35
 
36
  def english_to_hindi(sentence):
37
  translation_tokenizer.src_lang = "en_xx"
38
- encoded_hi = translation_tokenizer(article_hi, return_tensors="pt")
39
  generated_tokens = translation_model.generate(**encoded_hi, forced_bos_token_id=translation_tokenizer.lang_code_to_id["hi_IN"] )
40
  return (translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True))
41
 
42
 
43
  def hindi_to_english(sentence):
44
  translation_tokenizer.src_lang = "hi_IN"
45
- encoded_hi = translation_tokenizer(article_hi, return_tensors="pt")
46
  generated_tokens = translation_model.generate(**encoded_hi, forced_bos_token_id=translation_tokenizer.lang_code_to_id["en_XX"] )
47
  return (translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True))
48
 
 
35
 
36
  def english_to_hindi(sentence):
37
  translation_tokenizer.src_lang = "en_xx"
38
+ encoded_hi = translation_tokenizer(sentence, return_tensors="pt")
39
  generated_tokens = translation_model.generate(**encoded_hi, forced_bos_token_id=translation_tokenizer.lang_code_to_id["hi_IN"] )
40
  return (translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True))
41
 
42
 
43
  def hindi_to_english(sentence):
44
  translation_tokenizer.src_lang = "hi_IN"
45
+ encoded_hi = translation_tokenizer(sentence, return_tensors="pt")
46
  generated_tokens = translation_model.generate(**encoded_hi, forced_bos_token_id=translation_tokenizer.lang_code_to_id["en_XX"] )
47
  return (translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True))
48