diff --git a/src/models/emote_vision_model.py b/src/models/emote_vision_model.py
index f855486..a4fe2fb 100644
--- a/src/models/emote_vision_model.py
+++ b/src/models/emote_vision_model.py
@@ -5,7 +5,7 @@
 
 
 class EmoteVisionModel(nn.Module):
-    def __init__(self, embedding_size: int = 512, num_classes: int = 7):
+    def __init__(self, embedding_size: int = 512, num_classes: int = 4):
         super().__init__()
 
         # Pretrained ResNet50 backbone