add emotion classification

README.md

@@ -7,7 +7,7 @@ sdk: streamlit
 sdk_version: 1.10.0
 app_file: app.py
 pinned: false
-models: cardiffnlp/twitter-roberta-base-emoji
+models: mrm8488/t5-base-finetuned-emotion
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
-from emoji import get_emoji
+# from emoji import get_emoji
+from emotions import get_emotion
 
 st.title("I don't even know what this is yet")
 
@@ -7,6 +8,6 @@ name = st.text_input('Who are you?', 'Zeliboba')
 text = st.text_area('Submit your stories', '''Random symbols''')
 
 st.write('Here is your first text:', text)
-st.write(get_emoji(text))
+st.write(get_emotion(text))
 
 

emotions.py

@@ -0,0 +1,16 @@
+from transformers import AutoTokenizer, AutoModelWithLMHead
+
+tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-emotion")
+
+model = AutoModelWithLMHead.from_pretrained("mrm8488/t5-base-finetuned-emotion")
+
+
+def get_emotion(text):
+    input_ids = tokenizer.encode(text + '</s>', return_tensors='pt')
+
+    output = model.generate(input_ids=input_ids,
+                            max_length=2)
+
+    dec = [tokenizer.decode(ids) for ids in output]
+    label = dec[0]
+    return label