Duplicate from harish3110/emotion_detection

Co-authored-by: Harish Vadlamani <harish3110@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
+*.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
+*.wasm 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,13 @@
+---
+title: Emotion Detection
+emoji: 🌖
+colorFrom: blue
+colorTo: indigo
+sdk: gradio
+sdk_version: 3.0.19
+app_file: app.py
+pinned: false
+duplicated_from: harish3110/emotion_detection
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,32 @@
+from speechbrain.pretrained.interfaces import foreign_class
+import gradio as gr
+
+import warnings
+warnings.filterwarnings("ignore")
+
+# Loading the speechbrain emotion detection model
+learner = foreign_class(
+    source="speechbrain/emotion-recognition-wav2vec2-IEMOCAP",
+    # savedir="/home/harish3110/SeaWord/emotion/nbs/pretrained_models/CustomEncoderWav2vec2Classifier--8353113631630090076",
+    pymodule_file="custom_interface.py", 
+    classname="CustomEncoderWav2vec2Classifier"
+)
+
+# Building prediction function for gradio
+emotion_dict = {
+    'sad': 'Sad', 
+    'hap': 'Happy',
+    'ang': 'Anger',
+    'neu': 'Neutral'
+}
+
+def predict_emotion(audio):
+    out_prob, score, index, text_lab = learner.classify_file(audio.name)
+    return emotion_dict[text_lab[0]]
+
+# Loading gradio interface
+inputs = gr.inputs.Audio(label="Input Audio", type="file")
+outputs = "text"
+title = "Emotion Detection"
+description = "Gradio demo for Emotion Detection. To use it, simply upload your audio, or click one of the examples to load them. Read more at the links below."
+gr.Interface(predict_emotion, inputs, outputs, title=title, description=description).launch()

requirements.txt

@@ -0,0 +1,3 @@
+speechbrain
+torchaudio
+transformers