Update README.md

README.md

@@ -83,7 +83,7 @@ id2label = {int(k): v for k, v in model.config.id2label.items()}
 top5 = probs.topk(5)
 for score, idx in zip(top5.values, top5.indices):
 print(f"Label {idx.item()} (celeb_id {id2label[idx.item()]}): {score:.3f}")
-
+```
 
 ## Training Details