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

Create app1.py

Browse files
Files changed (1) hide show
  1. app1.py +13 -0
app1.py ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+