Update app.py

app.py

@@ -34,21 +34,17 @@ translation_tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-lar
 
 
 def english_to_hindi(sentence):
-  translation_tokenizer.src_lang = "en_xx"
-  encoded_hi = translation_tokenizer(article_hi, return_tensors="pt")
-  generated_tokens = translation_model.generate(
-      **encoded_hi, forced_bos_token_id=translation_tokenizer.lang_code_to_id["hi_IN"] )
-
-  return (translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True))
+    translation_tokenizer.src_lang = "en_xx"
+    encoded_hi = translation_tokenizer(article_hi, return_tensors="pt")
+    generated_tokens = translation_model.generate(**encoded_hi, forced_bos_token_id=translation_tokenizer.lang_code_to_id["hi_IN"] )
+    return (translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True))
 
 
 def hindi_to_english(sentence):
-  translation_tokenizer.src_lang = "hi_IN"
-  encoded_hi = translation_tokenizer(article_hi, 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))
+    translation_tokenizer.src_lang = "hi_IN"
+    encoded_hi = translation_tokenizer(article_hi, 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))
 
 
 user_session_id = uuid.uuid4()