Update app.py
Browse files
app.py
CHANGED
@@ -31,20 +31,22 @@ languages_list = [("Gujarati", "gu_IN"), ('Hindi',"hi_IN") , ("Bengali","bn_IN")
|
|
31 |
lang_global = ''
|
32 |
def intitalize_lang(language):
|
33 |
lang_global = language
|
34 |
-
print(lang_global)
|
35 |
|
36 |
def english_to_indian(sentence):
|
37 |
-
print (lang_global)
|
38 |
translation_tokenizer.src_lang = "en_xx"
|
39 |
encoded_hi = translation_tokenizer(sentence, return_tensors="pt")
|
40 |
-
generated_tokens = translation_model.generate(**encoded_hi,
|
|
|
41 |
return (translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True))
|
42 |
|
43 |
|
44 |
def indian_to_english(sentence):
|
45 |
translation_tokenizer.src_lang = lang_global
|
46 |
encoded_hi = translation_tokenizer(sentence, return_tensors="pt")
|
47 |
-
generated_tokens = translation_model.generate(**encoded_hi,
|
|
|
48 |
return (translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True))
|
49 |
|
50 |
|
|
|
31 |
lang_global = ''
|
32 |
def intitalize_lang(language):
|
33 |
lang_global = language
|
34 |
+
print("intitalize_lang"+lang_global)
|
35 |
|
36 |
def english_to_indian(sentence):
|
37 |
+
print ("english_to_indian"+lang_global)
|
38 |
translation_tokenizer.src_lang = "en_xx"
|
39 |
encoded_hi = translation_tokenizer(sentence, return_tensors="pt")
|
40 |
+
generated_tokens = translation_model.generate(**encoded_hi,
|
41 |
+
forced_bos_token_id=translation_tokenizer.lang_code_to_id[lang_global] )
|
42 |
return (translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True))
|
43 |
|
44 |
|
45 |
def indian_to_english(sentence):
|
46 |
translation_tokenizer.src_lang = lang_global
|
47 |
encoded_hi = translation_tokenizer(sentence, return_tensors="pt")
|
48 |
+
generated_tokens = translation_model.generate(**encoded_hi,
|
49 |
+
forced_bos_token_id=translation_tokenizer.lang_code_to_id["en_XX"] )
|
50 |
return (translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True))
|
51 |
|
52 |
|