06ea51c
1
2
3
4
5
6
7
8
9
10
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)
