versae's picture
Add PyTorch model
c343f62