Plachta's picture
Update ONNXVITS_infer.py
9aeb33e