Ojimi
/

vit-anime-caption

Image Classification

Model card Files Files and versions

Ojimi commited on Jan 7, 2024

Commit

994870c

·

1 Parent(s): d63798c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -19,7 +19,7 @@ from transformers import ViTImageProcessor, ViTForImageClassification
 import torch
 from PIL import Image
-model_name_or_path = "vit-anime-base/"
 processor = ViTImageProcessor.from_pretrained(model_name_or_path)
 model = ViTForImageClassification.from_pretrained(model_name_or_path)
 threshold = 0.3

 import torch
 from PIL import Image
+model_name_or_path = "Ojimi/vit-anime-caption"
 processor = ViTImageProcessor.from_pretrained(model_name_or_path)
 model = ViTForImageClassification.from_pretrained(model_name_or_path)
 threshold = 0.3