lrzjason commited on
Commit
6114cab
1 Parent(s): 38514c6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -1
README.md CHANGED
@@ -39,4 +39,7 @@ with torch.no_grad():
39
  # model predicts one of the 1000 ImageNet classes
40
  predicted_label = logits.argmax(-1).item()
41
  print(f'predicted_label: {model.config.id2label[predicted_label]}')
42
- ```
 
 
 
 
39
  # model predicts one of the 1000 ImageNet classes
40
  predicted_label = logits.argmax(-1).item()
41
  print(f'predicted_label: {model.config.id2label[predicted_label]}')
42
+ ```
43
+
44
+ Using this dataset:
45
+ https://huggingface.co/datasets/animelover/genshin-impact-images