vishwask commited on
Commit
2a83935
·
verified ·
1 Parent(s): 7570d2b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -2
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" ,"bn_IN", "ml_IN", "mr_IN", "ta_IN", "te_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.Radio(languages_list, label="Languages", value = list_llm_simple[1], type="index", info="Choose your language")
 
 
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