tareesh commited on
Commit
f942882
1 Parent(s): 8a8fd5c

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -25
app.py DELETED
@@ -1,25 +0,0 @@
1
- import streamlit as st
2
-
3
- pip install transformers -U -q
4
-
5
- pip install sentencepiece
6
-
7
- pip freeze | grep transformers
8
-
9
- from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
10
-
11
- text = st.text_area('enter some text:')
12
-
13
- model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-one-to-many-mmt")
14
- tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-one-to-many-mmt", src_lang="en_XX")
15
-
16
-
17
- if text:
18
- generated_tokens = model.generate(
19
- **model_inputs,
20
- forced_bos_token_id=tokenizer.lang_code_to_id["hi_IN"]
21
- )
22
- tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
23
- st.json(translation)
24
-
25
-