himani's picture
add pytorch model
41ac0ae