Update app.py

app.py

@@ -81,7 +81,7 @@ def predict_emotion(video_path):
     return last_frame, audio_path, predicted_label
 
 def predict_emotion_gradio(video_path):
-    emotion_dict = {0: 'neutral', 1: 'calm', 2: 'happy', 3: 'sad', 4: 'angry', 5: 'fearful'}
+    emotion_dict = {0: 'neutral', 1: 'calm', 2: 'happy', 3: 'sad', 4: 'angry', 5: 'fearful', 6: 'disgust', 7: 'surprised'}
     last_frame, audio_path, predicted_label = predict_emotion(video_path)
     predicted_emotion = emotion_dict[predicted_label]
     return last_frame, audio_path, predicted_emotion