Update app.py
Browse files
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(
|
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(
|
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 |
|