lrzjason
/

anime_portrait_vit

Image Classification

Inference Endpoints

Model card Files Files and versions Community

lrzjason commited on Feb 16

Commit

6114cab

•

1 Parent(s): 38514c6

Update README.md

Files changed (1) hide show

README.md +4 -1

README.md CHANGED Viewed

@@ -39,4 +39,7 @@ with torch.no_grad():
 # model predicts one of the 1000 ImageNet classes
 predicted_label = logits.argmax(-1).item()
 print(f'predicted_label: {model.config.id2label[predicted_label]}')
-```

 # model predicts one of the 1000 ImageNet classes
 predicted_label = logits.argmax(-1).item()
 print(f'predicted_label: {model.config.id2label[predicted_label]}')
+```
+Using this dataset:
+https://huggingface.co/datasets/animelover/genshin-impact-images