bhadresh-savani's picture
Added pytorch model
0c24d02