Update app.py
Browse files
app.py
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
import streamlit as st
|
2 |
-
from transformers import
|
3 |
import sentencepiece
|
4 |
|
5 |
text = st.text_area('Enter the text:')
|
6 |
-
model =
|
7 |
tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-one-to-many-mmt", src_large="en_XX")
|
8 |
|
9 |
if text:
|
|
|
1 |
import streamlit as st
|
2 |
+
from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
|
3 |
import sentencepiece
|
4 |
|
5 |
text = st.text_area('Enter the text:')
|
6 |
+
model = MBartForConditionalGenerationneration.from_pretrained("facebook/mbart-large-50-one-to-many-mmt")
|
7 |
tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-one-to-many-mmt", src_large="en_XX")
|
8 |
|
9 |
if text:
|