examples/example_usage.py

import torch
from model import *
from inference import load_model

if __name__ == "__main__":
    model = load_model("model.pt")
    x = torch.randn(1, 3, 224, 224)
    y = model(x)
    print("Example output:", y.shape)