dummy.py - Dummy model architecturevanilla.py - Vanilla model architecturedummyweights - Trained weights for dummy modelvanillaweights - Trained weights for vanilla model1import torch
2
3# Load pretrained vanilla model
4model = torch.hub.load('your-username/A4_4312645_ziyad', 'vanilla', pretrained=True)
5model.eval()