Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)
|