Accept image data in MNIST cpp example instead of .pt file

### 🚀 The feature

currently we only accept .pt file

### Motivation, pitch

requires less steps to try it out + image compression on transfer

### Alternatives

_No response_

### Additional context

_No response_