tareesh commited on
Commit
5f6740e
·
1 Parent(s): b753c1e

Update app1.py

Browse files
Files changed (1) hide show
  1. app1.py +10 -5
app1.py CHANGED
@@ -1,13 +1,18 @@
1
  import streamlit as st
2
- import transformers import MBartForConditionalGeneration, MBart50TokenizerFast
 
 
3
 
4
  model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-one-to-many-mmt")
5
- tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-one-to-many-mmt", src_large="en_XX")
6
 
7
- text = st.text_area('enter some text:')
8
 
9
  if text:
10
- generated_tokens = model.generate(**model_inputs, forced_bos_token_id=tokenizer.lang_code_to_id["hi_IN"])
11
- translation = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
 
 
 
12
  st.json(translation)
13
 
 
 
1
  import streamlit as st
2
+ from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
3
+
4
+ text = st.text_area('enter some text:')
5
 
6
  model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-one-to-many-mmt")
7
+ tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-one-to-many-mmt", src_lang="en_XX")
8
 
 
9
 
10
  if text:
11
+ generated_tokens = model.generate(
12
+ **model_inputs,
13
+ forced_bos_token_id=tokenizer.lang_code_to_id["hi_IN"]
14
+ )
15
+ tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
16
  st.json(translation)
17
 
18
+