Update README.md

README.md

@@ -39,7 +39,7 @@ outputs = model(images, alpha=0.7)
 labels, scores = outputs["vocabularies"][0], outputs["scores"][0]
 
 # print the top 5 most likely labels for the image
-values, indices = scores.sort(dim=-1, descending=True)
+values, indices = scores.topk(3)
 print("\nTop predictions:\n")
 for value, index in zip(values, indices):
     print(f"{labels[index]:>16s}: {100 * value.item():.2f}%")