trained_model_weights.pth 45 MB