Adjustments to make it running on Colab

[timspilak]

It wasn't running in colab, so I had to make the following adjustments:

inference.py line 40 &41:

    'resnet': resnet18(weights='IMAGENET1K_V1'),
    'resnet18': resnet18(weights='IMAGENET1K_V1'),

inference.py line 74:

checkpoint = torch.load(reg_weights, weights_only=True)`

library/Math.py", line 163:

A = np.zeros([4,3], dtype=np.float64)