ruchitoshniwal commited on
Commit
4b616ca
โ€ข
1 Parent(s): 7c1f322

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -8,9 +8,9 @@ from transformers import pipeline
8
 
9
  # Load pipeline
10
  pipe_lr = joblib.load(open("emotion_detector_pipe_lr.pkl","rb"))
11
- #emoroberta_emotion_classifier = pipeline(
12
- # "text-classification", model="arpanghoshal/EmoRoBERTa", return_all_scores=True
13
- #)
14
 
15
  # Emojis
16
  emotions_emoji_dict = {"anger":"๐Ÿ˜ ","disgust":"๐Ÿคฎ", "fear":"๐Ÿ˜จ๐Ÿ˜ฑ", "joy":"๐Ÿค—", "neutral":"๐Ÿ˜", "sadness":"๐Ÿ˜”", "shame":"๐Ÿ˜ณ", "surprise":"๐Ÿ˜ฎ"}
 
8
 
9
  # Load pipeline
10
  pipe_lr = joblib.load(open("emotion_detector_pipe_lr.pkl","rb"))
11
+ emoroberta_emotion_classifier = pipeline(
12
+ "text-classification", model="arpanghoshal/EmoRoBERTa", return_all_scores=True
13
+ )
14
 
15
  # Emojis
16
  emotions_emoji_dict = {"anger":"๐Ÿ˜ ","disgust":"๐Ÿคฎ", "fear":"๐Ÿ˜จ๐Ÿ˜ฑ", "joy":"๐Ÿค—", "neutral":"๐Ÿ˜", "sadness":"๐Ÿ˜”", "shame":"๐Ÿ˜ณ", "surprise":"๐Ÿ˜ฎ"}