Plachta's picture
Update ONNXVITS_infer.py
eac9bb8