Could you please share a version with the model definition for the pretrained model present? Really appreciate the neat work! TIA