umang018 commited on
Commit
c4e8e59
·
verified ·
1 Parent(s): 243a4d7

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=false)
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=True)
8
 
9
  # Streamlit app
10
  def main():