Duplicate from danielferreira/emotion-text-classification

Co-authored-by: Daniel Ferreira <danielferreira@users.noreply.huggingface.co>

.gitattributes

@@ -0,0 +1,27 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bin.* filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zstandard filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,38 @@
+---
+title: Emotion Text Classification
+emoji: 😊
+colorFrom: yellow
+colorTo: red
+sdk: gradio
+app_file: app.py
+pinned: false
+duplicated_from: danielferreira/emotion-text-classification
+---
+
+# Configuration
+
+`title`: _string_  
+Display title for the Space
+
+`emoji`: _string_  
+Space emoji (emoji-only character allowed)
+
+`colorFrom`: _string_  
+Color for Thumbnail gradient (red, yellow, green, blue, indigo, purple, pink, gray)
+
+`colorTo`: _string_  
+Color for Thumbnail gradient (red, yellow, green, blue, indigo, purple, pink, gray)
+
+`sdk`: _string_  
+Can be either `gradio` or `streamlit`
+
+`sdk_version` : _string_  
+Only applicable for `streamlit` SDK.  
+See [doc](https://hf.co/docs/hub/spaces) for more info on supported versions.
+
+`app_file`: _string_  
+Path to your main application file (which contains either `gradio` or `streamlit` Python code).  
+Path is relative to the root of the repository.
+
+`pinned`: _boolean_  
+Whether the Space stays on top of your list.

app.py

@@ -0,0 +1,53 @@
+import gradio as gr
+from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
+
+
+class EmotionClassifier:
+    def __init__(self, model_name: str):
+        self.model = AutoModelForSequenceClassification.from_pretrained(model_name)
+        self.tokenizer = AutoTokenizer.from_pretrained(model_name)
+        self.pipeline = pipeline(
+            "text-classification",
+            model=self.model,
+            tokenizer=self.tokenizer,
+            return_all_scores=True,
+        )
+
+    def predict(self, input_text: str):
+        pred = self.pipeline(input_text)[0]
+        result = {
+            "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 result
+
+
+def main():
+    model = EmotionClassifier("bhadresh-savani/bert-base-uncased-emotion")
+    iface = gr.Interface(
+        fn=model.predict,
+        inputs=gr.inputs.Textbox(
+            lines=3,
+            placeholder="Type a phrase that has some emotion",
+            label="Input Text",
+        ),
+        outputs="label",
+        title="Emotion Classification",
+        examples=[
+            "I get so down when I'm alone",
+            "I believe that today everything will work out",
+            "It was so dark there I was afraid to go",
+            "I loved the gift you gave me",
+            "I was very surprised by your presentation.",
+        ],
+    )
+
+    iface.launch()
+
+
+if __name__ == "__main__":
+    main()

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