umang018 commited on
Commit
fb24256
·
verified ·
1 Parent(s): db62c31

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
4
  # Load the emotion classification model
5
  @st.cache(allow_output_mutation=True)
6
  def load_model():
7
- return pipeline('text-classification', model='SamLowe/roberta-base-go_emotions', return_all_scores=True)
8
 
9
  # Streamlit app
10
  def main():
 
4
  # Load the emotion classification model
5
  @st.cache(allow_output_mutation=True)
6
  def load_model():
7
+ return pipeline('text-classification', model='SamLowe/roberta-base-go_emotions', return_all_scores=false)
8
 
9
  # Streamlit app
10
  def main():