Update critical error of README.md (#1)

- Update critical error of README.md (15f1799451117dd109095b2744ce66584de0114d)


Co-authored-by: Jorge González <JvL359@users.noreply.huggingface.co>

README.md

@@ -146,5 +146,5 @@ with torch.no_grad():
     probs = torch.softmax(logits, dim=1)
     conf, pred = probs.max(dim=1)
 
-print(f"Prediction: {classes[str(pred.item())]} (Confidence: {conf.item()*100:.1f}%)")
+print(f"Prediction: {classes[int(pred.item())]} (Confidence: {conf.item()*100:.1f}%)")
 ```