Update model.py

model.py

@@ -32,7 +32,7 @@ class EmotionPredictor:
     def __init__(self):
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-        with open("classes.pkl", "rb") as f:
+        with open("classes.pkl", "wb") as f:
             self.classes = pickle.load(f)
 
         self.model = FaceClassifier(len(self.classes))