xujinheng666 commited on
Commit
32cfaa6
·
verified ·
1 Parent(s): 6b3d9d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def load_models():
18
 
19
  st.session_state.summary_pipe = pipeline("summarization", model="Falconsai/text_summarization")
20
 
21
- st.session_state.rating_pipe = pipeline("text-classification", model="cardiffnlp/twitter-roberta-base-sentiment-latest")
22
 
23
  def transcribe_audio(audio_path):
24
  pipe = st.session_state.transcription_pipe
 
18
 
19
  st.session_state.summary_pipe = pipeline("summarization", model="Falconsai/text_summarization")
20
 
21
+ st.session_state.rating_pipe = pipeline("text-classification", model="distilbert/distilbert-base-uncased-finetuned-sst-2-english")
22
 
23
  def transcribe_audio(audio_path):
24
  pipe = st.session_state.transcription_pipe