TheRealAIGuy's picture
Minor fix in inference.py
060c9f5