THP2903 commited on
Commit
3fdae12
·
verified ·
1 Parent(s): 28b224f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,6 +3,7 @@ import torch as pt
3
  import torchaudio
4
  import cv2
5
  import os
 
6
  import numpy as np
7
  import tensorflow as tf
8
  from tensorflow.keras.models import load_model
@@ -154,7 +155,7 @@ def predict_emotion(video_path, audio_path):
154
 
155
 
156
 
157
- predicted_label = predict_emotion(video_path)
158
  emotion_dict = {0: 'neutral', 1: 'calm', 2: 'happy', 3: 'sad', 4: 'angry', 5: 'fearful'}
159
  predicted_emotion = emotion_dict[predicted_label]
160
  print("Predicted Emotion: ", predicted_emotion)
 
3
  import torchaudio
4
  import cv2
5
  import os
6
+
7
  import numpy as np
8
  import tensorflow as tf
9
  from tensorflow.keras.models import load_model
 
155
 
156
 
157
 
158
+ predicted_label = predict_emotion(video_path, audio_path)
159
  emotion_dict = {0: 'neutral', 1: 'calm', 2: 'happy', 3: 'sad', 4: 'angry', 5: 'fearful'}
160
  predicted_emotion = emotion_dict[predicted_label]
161
  print("Predicted Emotion: ", predicted_emotion)