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