Make model of pytorch_stream.ipynb more realistic

[Leengit]

The current model may not be so appropriate and it is not trained. Something similar to the model used in tensorflow_stream.ipynb would be good but, more importantly, the model should be a trained model. The goal is for the Jupyter lab to show the user that histomics_stream can produce meaningful output with a meaningful model.