shubhankargaur commited on
Commit
c689e33
1 Parent(s): d0daec4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,10 +8,10 @@ tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-one-to
8
 
9
  def get_input(text):
10
  models_input = tokenizer(text,return_tensors="pt")
11
- generated_tokens = model.generate(**models_input,forced_bos_token_id=tokenizer.lang_code_to_id["ta_IN"])
12
  translation = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
13
  return translation
14
 
15
- iface = gr.Interface(fn=get_input,inputs="text",outputs="text", title = "English to Tamil Translator",description="Get Tamil translation for your text in English")
16
 
17
  iface.launch()
 
8
 
9
  def get_input(text):
10
  models_input = tokenizer(text,return_tensors="pt")
11
+ generated_tokens = model.generate(**models_input,forced_bos_token_id=tokenizer.lang_code_to_id["hi_IN"])
12
  translation = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
13
  return translation
14
 
15
+ iface = gr.Interface(fn=get_input,inputs="text",outputs="text", title = "English to Hindi Translator",description="Translate the text given in English to Hindi")
16
 
17
  iface.launch()