Fix example
#1
by Ccellado - opened
Example not working model expects int
label = id2label[str(pred.item())] -> label = id2label[pred.item()]
can you post a PR? Thanks
Example not working model expects int
label = id2label[str(pred.item())] -> label = id2label[pred.item()]
can you post a PR? Thanks