Plachta's picture
Update ONNXVITS_infer.py
4b86b18