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