Jash Mevada commited on
Commit
1f06aea
·
1 Parent(s): 1601426

updated title.

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +3 -3
src/streamlit_app.py CHANGED
@@ -80,10 +80,10 @@ def predict_emotions(texts: List[str], threshold: float) -> pd.DataFrame:
80
  return df
81
 
82
 
83
- st.title("Multi-label Emotion Classifier (Fine-tuned DistilBERT)")
84
  st.write(
85
- "Uploads from Hugging Face (`JashMevada/emotion-classifier`) are used to run the fine-tuned head "
86
- "from the notebook’s `FineTuneModel`. Adjust the threshold to suit your precision/recall trade-off."
87
  )
88
 
89
  with st.sidebar:
 
80
  return df
81
 
82
 
83
+ st.title("Multi-label Emotion Classifier (Fine-tuned RoBERTa)")
84
  st.write(
85
+ "Uploads from Hugging Face (`JashMevada/emotion-classifier`) are used to run the fine-tuned head."
86
+ "Adjust the threshold to suit your precision/recall trade-off."
87
  )
88
 
89
  with st.sidebar: