trying to add a pytorch save file instead of a pkl file as huggingface spaces doesn't want to load the pickle file

10faea5