config.json

{
  "model_type": "swin-transformer",
  "architecture": "CustomSwinTransformer",
  "num_classes": 7,
  "emotions": [
    "Angry",
    "Disgust",
    "Fear",
    "Happy",
    "Neutral",
    "Sad",
    "Surprise"
  ],
  "input_size": [
    224,
    224
  ],
  "framework": "pytorch",
  "license": "mit"
}