Update app.py

app.py

@@ -28,20 +28,19 @@ languages_list = [("Gujarati", "gu_IN"), ('Hindi',"hi_IN") , ("Bengali","bn_IN")
                   ("Marathi","mr_IN"), ("Tamil","ta_IN"), ("Telugu","te_IN")]
 
 
+lang_global = ''
 def intitalize_lang(language):
-    return language
+    lang_global = language
 
 def english_to_indian(sentence):
-    lang = intitalize_lang()
     translation_tokenizer.src_lang = "en_xx"
     encoded_hi = translation_tokenizer(sentence, return_tensors="pt")
-    generated_tokens = translation_model.generate(**encoded_hi, forced_bos_token_id=translation_tokenizer.lang_code_to_id[lang] )
+    generated_tokens = translation_model.generate(**encoded_hi, forced_bos_token_id=translation_tokenizer.lang_code_to_id[lang_global] )
     return (translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True))
 
 
 def indian_to_english(sentence):
-    lang = intitalize_lang()
-    translation_tokenizer.src_lang = lang
+    translation_tokenizer.src_lang = lang_global
     encoded_hi = translation_tokenizer(sentence, return_tensors="pt")
     generated_tokens = translation_model.generate(**encoded_hi, forced_bos_token_id=translation_tokenizer.lang_code_to_id["en_XX"] )
     return (translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True))