Update app.py

app.py

@@ -33,6 +33,23 @@ translation_model = MBartForConditionalGeneration.from_pretrained("facebook/mbar
 translation_tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
 
 
+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))
+
+
+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))
+
 
 user_session_id = uuid.uuid4()
 
@@ -170,7 +187,7 @@ if prompt := st.chat_input("How can I help you today?"):
             message_placeholder = st.empty()
             full_response = ""
             message_history = "\n".join(list(get_message_history())[-3:])
-            result = qa_chain(prompt)
+                result = qa_chain(hindi_to_english(prompt))
             output = [result['result']]
 
     def generate_pdf():