Hi MisBah-Awan,
You had replace the model by the dictionary. You should change the line 20 like this I think:
model = model()
model.load_state_dict(torch.load('last_brain1.pth')) #line to change
model.eval()

Let me know if you’re still stuck.