Update app.py
Browse files
app.py
CHANGED
|
@@ -30,9 +30,11 @@ languages_list = [("Gujarati", "gu_IN"), ('Hindi',"hi_IN") , ("Bengali","bn_IN")
|
|
| 30 |
|
| 31 |
lang_global = ''
|
| 32 |
def intitalize_lang(language):
|
|
|
|
| 33 |
lang_global = language
|
| 34 |
|
| 35 |
def english_to_indian(sentence):
|
|
|
|
| 36 |
translation_tokenizer.src_lang = "en_xx"
|
| 37 |
encoded_hi = translation_tokenizer(sentence, return_tensors="pt")
|
| 38 |
generated_tokens = translation_model.generate(**encoded_hi, forced_bos_token_id=translation_tokenizer.lang_code_to_id[lang_global] )
|
|
|
|
| 30 |
|
| 31 |
lang_global = ''
|
| 32 |
def intitalize_lang(language):
|
| 33 |
+
print(language)
|
| 34 |
lang_global = language
|
| 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, forced_bos_token_id=translation_tokenizer.lang_code_to_id[lang_global] )
|