Update app.py
Browse files
app.py
CHANGED
@@ -24,7 +24,10 @@ from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
|
|
24 |
translation_model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
|
25 |
translation_tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
|
26 |
|
27 |
-
languages_list = ["gu_IN", "hi_IN" ,"
|
|
|
|
|
|
|
28 |
|
29 |
def english_to_indian(sentence):
|
30 |
translation_tokenizer.src_lang = "en_xx"
|
@@ -39,6 +42,8 @@ def indian_to_english(sentence):
|
|
39 |
generated_tokens = translation_model.generate(**encoded_hi, forced_bos_token_id=translation_tokenizer.lang_code_to_id["en_XX"] )
|
40 |
return (translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True))
|
41 |
|
|
|
|
|
42 |
|
43 |
llm_model = "mistralai/Mistral-7B-Instruct-v0.2"
|
44 |
|
@@ -268,7 +273,9 @@ def demo():
|
|
268 |
qachain_btn = gr.Button("Initialize question-answering chain...")
|
269 |
|
270 |
with gr.Row():
|
271 |
-
lang_btn = gr.
|
|
|
|
|
272 |
|
273 |
chatbot = gr.Chatbot(height=300)
|
274 |
|
|
|
24 |
translation_model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
|
25 |
translation_tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
|
26 |
|
27 |
+
languages_list = [("Gujarati", "gu_IN"), ('Hindi',"hi_IN") , ("Bengali","bn_IN"), ("Malayalam","ml_IN"),
|
28 |
+
("Marathi","mr_IN"), ("Tamil","ta_IN"), ("Telugu","te_IN")]
|
29 |
+
|
30 |
+
|
31 |
|
32 |
def english_to_indian(sentence):
|
33 |
translation_tokenizer.src_lang = "en_xx"
|
|
|
42 |
generated_tokens = translation_model.generate(**encoded_hi, forced_bos_token_id=translation_tokenizer.lang_code_to_id["en_XX"] )
|
43 |
return (translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True))
|
44 |
|
45 |
+
def intitalize_lang(language):
|
46 |
+
|
47 |
|
48 |
llm_model = "mistralai/Mistral-7B-Instruct-v0.2"
|
49 |
|
|
|
273 |
qachain_btn = gr.Button("Initialize question-answering chain...")
|
274 |
|
275 |
with gr.Row():
|
276 |
+
lang_btn = gr.Dropdown(languages_list, label="Languages", value = languages_list[1],
|
277 |
+
type="value", info="Choose your language",interactive = True)
|
278 |
+
#lang_btn.select(intitalize_lang, inputs)
|
279 |
|
280 |
chatbot = gr.Chatbot(height=300)
|
281 |
|