Fix: Update torch.load to use weights_only=True to prevent security w…

[yaslack]

Fix: Update torch.load to use weights_only=True to prevent security warning

Changes Made:

In this pull request, I updated the torch.load function call to use weights_only=True in order to prevent a security warning related to the unpickling process.

Code Change:

- checkpoint = torch.load(fp, map_location=device)
+ checkpoint = torch.load(fp, map_location=device, weights_only=True)