xujinheng666 commited on
Commit
dd37445
·
verified ·
1 Parent(s): 01ddcf5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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("text-summarization", model="facebook/bart-large-cnn")
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