mattia-re-learn's picture
Update code/inference.py
f808930 verified