Fix argmax/softmax bug and normalization mismatch in inference (preprocess.py)

c5c411b

verified

suhanii23 commited on 25 days ago