Update app.py

app.py

@@ -42,8 +42,14 @@ def indian_to_english(sentence):
     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))
 
-#def intitalize_lang(language):
-    
+def intitalize_lang(language):
+    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[language] )
+    x = translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
+    print(language)
+    return x
+
 
 llm_model = "mistralai/Mistral-7B-Instruct-v0.2"
 
@@ -250,9 +256,6 @@ def demo():
             with gr.Row():
                 slider_chunk_overlap = gr.Slider(value=256, label="Chunk overlap", info="Chunk overlap", interactive=False, visible=False)
 
-            
-
-
         with gr.Accordion("Advanced options - LLM model", open=False, visible=False):
             with gr.Row():
                 slider_temperature = gr.Slider(value = 0.1,visible=False)
@@ -261,8 +264,6 @@ def demo():
             with gr.Row():
                 slider_topk = gr.Slider(value = 3, visible=False)
 
-
-
         with gr.Row():
             db_progress = gr.Textbox(label="Vector database initialization", value="None", visible=True)
         with gr.Row():
@@ -274,7 +275,7 @@ def demo():
 
         with gr.Row():
             lang_btn = gr.Dropdown(languages_list, label="Languages", value = languages_list[1],
-                                type="value", info="Choose your language",interactive = True)
+                                type="value", info="Choose your language",interactive = True).select(intitalize_lang)
         #lang_btn.select(intitalize_lang, inputs)
         
         chatbot = gr.Chatbot(height=300)