Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ def load_models():
|
|
16 |
st.session_state.translation_tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-zh-en")
|
17 |
st.session_state.translation_model = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-zh-en")
|
18 |
|
19 |
-
st.session_state.summary_pipe = pipeline("
|
20 |
|
21 |
st.session_state.rating_pipe = pipeline("text-classification", model="cardiffnlp/twitter-roberta-base-sentiment-latest")
|
22 |
|
|
|
16 |
st.session_state.translation_tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-zh-en")
|
17 |
st.session_state.translation_model = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-zh-en")
|
18 |
|
19 |
+
st.session_state.summary_pipe = pipeline("summarization", model="facebook/bart-large-cnn")
|
20 |
|
21 |
st.session_state.rating_pipe = pipeline("text-classification", model="cardiffnlp/twitter-roberta-base-sentiment-latest")
|
22 |
|