Added emotion classification

README.md

@@ -1,8 +1,8 @@
 ---
 title: Emotion Text Classification
-emoji: 🦀
-colorFrom: purple
-colorTo: blue
+emoji: 😊
+colorFrom: yellow
+colorTo: red
 sdk: gradio
 app_file: app.py
 pinned: false

app.py

@@ -1,7 +1,23 @@
 import gradio as gr
+from transformers import pipeline
 
-def greet(name):
-    return "Hello " + name + "!!"
+pipeline = pipeline("text-classification", model="bhadresh-savani/bert-base-uncased-emotion", return_all_scores=True)
+
+def predict(input_text):
+  preds = pipeline(input_text)
+  pred= preds[0]
+  res = {"Sadness 😭": pred[0]["score"],
+         "Joy 😂": pred[1]["score"],
+         "Love 😍": pred[2]["score"],
+         "Anger 😠": pred[3]["score"],
+         "Fear 😨": pred[4]["score"],
+         "Surprise 😲": pred[5]["score"],
+        }
+  return res
+
+iface = gr.Interface(fn = predict, 
+                     inputs = "text", 
+                     outputs = gr.outputs.Label(num_top_classes=None, type="auto", label=None),
+                     title = 'Sentiment Analysis')
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
 iface.launch()

requirements.txt

@@ -0,0 +1,74 @@
+analytics-python==1.4.0
+backcall==0.2.0
+backoff==1.10.0
+bcrypt==3.2.0
+certifi==2021.10.8
+cffi==1.15.0
+charset-normalizer==2.0.8
+click==8.0.3
+cryptography==36.0.0
+cycler==0.11.0
+debugpy==1.5.1
+decorator==5.1.0
+entrypoints==0.3
+ffmpy==0.3.0
+filelock==3.4.0
+Flask==2.0.2
+Flask-CacheBuster==1.0.0
+Flask-Cors==3.0.10
+Flask-Login==0.5.0
+fonttools==4.28.2
+gradio==2.4.6
+huggingface-hub==0.1.2
+idna==3.3
+ipykernel==6.5.1
+ipython==7.30.0
+itsdangerous==2.0.1
+jedi==0.18.1
+Jinja2==3.0.3
+joblib==1.1.0
+jupyter-client==7.1.0
+jupyter-core==4.9.1
+kiwisolver==1.3.2
+markdown2==2.4.1
+MarkupSafe==2.0.1
+matplotlib==3.5.0
+matplotlib-inline==0.1.3
+monotonic==1.6
+nest-asyncio==1.5.1
+numpy==1.21.4
+packaging==21.3
+pandas==1.3.4
+paramiko==2.8.1
+parso==0.8.2
+pexpect==4.8.0
+pickleshare==0.7.5
+Pillow==8.4.0
+prompt-toolkit==3.0.23
+ptyprocess==0.7.0
+pycparser==2.21
+pycryptodome==3.11.0
+pydub==0.25.1
+Pygments==2.10.0
+PyNaCl==1.4.0
+pyparsing==3.0.6
+python-dateutil==2.8.2
+pytz==2021.3
+PyYAML==6.0
+pyzmq==22.3.0
+regex==2021.11.10
+requests==2.26.0
+sacremoses==0.0.46
+setuptools-scm==6.3.2
+six==1.16.0
+tokenizers==0.10.3
+tomli==1.2.2
+torch==1.10.0
+tornado==6.1
+tqdm==4.62.3
+traitlets==5.1.1
+transformers==4.12.5
+typing-extensions==4.0.0
+urllib3==1.26.7
+wcwidth==0.2.5
+Werkzeug==2.0.2