Add onnx models to prediction

[mrvollger]

This simplifies the pytorch models to use onnx which allows me to use other pure rust backends.