app.py updated

app.py

@@ -6,13 +6,13 @@ from radar_chart import radar_factory
 
 from keras.models import load_model
 import os
-import joblib
+#import joblib
 import numpy as np
 import soundfile as sf
 
 
-model = joblib.load('Speech_Emotion_Detection_Model.pkl')
-
+#model = joblib.load('Speech_Emotion_Detection_Model.pkl')
+model = load_model(os.path.join("lstm_all_four_complex.h5"))
 
 def convert_class_to_emotion(pred):
         """
@@ -28,12 +28,14 @@ def convert_class_to_emotion(pred):
         #                     6: 'disgust',
         #                     7: 'surprised'}
 
-        label_conversion = {'NEU':'neutral',
-                            'HAP':'happy',
-                            'SAD':'sad',
-                            'ANG':'angry',
-                            'FEA':'fearful',
-                            'DIS':'disgust'
+        label_conversion = {'neutral':'neutral',
+                            'calm': 'calm',
+                            'happy':'happy',
+                            'sad':'sad',
+                            'angry':'angry',
+                            'fearful':'fearful',
+                            'disgust':'disgust',
+                            'ps': 'surprised'
                            }
 
         return label_conversion[int(pred)]