from ultralyticsplus import YOLO, postprocess_classify_output
# load model
model = YOLO('Revrse/icon-labelling')
# set image
image = 'test/155.png'
# perform inference
prediction = model(image)
# observe results
print(prediction[0].names[prediction[0].probs.top1])