AeonAIRisk's picture
Upload load_poc.py with huggingface_hub
d3a42e0 verified
Raw
History Blame Contribute Delete
198 Bytes
import pickle, pathlib, sys
p=pathlib.Path(sys.argv[1]) if len(sys.argv)>1 else pathlib.Path(__file__).with_name('evil.pkl')
print('loading', p)
pickle.loads(p.read_bytes())
print('loaded ok')