SamanthaStorm commited on
Commit
e5bac56
·
verified ·
1 Parent(s): 8dc9e3c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ def get_emotion_profile(text):
43
  emotions = emotions[0]
44
  return {e['label'].lower(): round(e['score'], 3) for e in emotions}
45
  # Emotion model (no retraining needed)
46
- emotion_pipeline = hf_pipeline(
47
  "text-classification",
48
  model="j-hartmann/emotion-english-distilroberta-base",
49
  top_k=6,
 
43
  emotions = emotions[0]
44
  return {e['label'].lower(): round(e['score'], 3) for e in emotions}
45
  # Emotion model (no retraining needed)
46
+ emotion_pipeline = pipeline(
47
  "text-classification",
48
  model="j-hartmann/emotion-english-distilroberta-base",
49
  top_k=6,