models/nipun_model.py

import torch.nn as nn
from facenet_pytorch import InceptionResnetV1

def get_nipun_model():
    model = InceptionResnetV1(pretrained='vggface2', classify=True, num_classes=7)
    # Match Nipun's structure - he wrapped logits in Sequential
    model.logits = nn.Sequential(
        nn.Linear(512, 7)
    )
    return model