YAML Metadata
Warning:
empty or missing yaml metadata in repo card
(https://huggingface.co/docs/hub/model-cards#model-card-metadata)
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])