MarcusSu1216's picture
Update inference_main.py
b80e41a verified