Upload main.py

main.py

@@ -68,13 +68,13 @@ async def predict(file: UploadFile = File(...)):
 
     class_names = ['angry', 'happy', 'sad', 'surprised', 'neutral', 'disgusted', 'fearful']
     stress_levels = {
-        'angry': 'High',
-        'happy': 'Normal',
-        'sad': 'Low',
-        'surprised': 'Low',
-        'neutral': 'Low',
-        'disgusted': 'Low',
-        'fearful': 'Low'
+        'angry': 'high_stress',
+        'happy': 'neutral',
+        'sad': 'low_stress',
+        'surprised': 'low_stress',
+        'neutral': 'neutral',
+        'disgusted': 'low_stress',
+        'fearful': 'low_stress'
     }
 
     emotion = class_names[predicted_class]