Plachta's picture
Update ONNXVITS_infer.py
452888f