AIris-Channel
/

vit-gpt2-verifycode-caption

vision-encoder-decoder

image-captioning

Model card Files Files and versions

TheKOG commited on Aug 17, 2023

Commit

33325eb

·

1 Parent(s): 11e82ae

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -11,9 +11,9 @@ from transformers import VisionEncoderDecoderModel, ViTImageProcessor, AutoToken
 import torch
 from PIL import Image
-model = VisionEncoderDecoderModel.from_pretrained("TheKOG/vit-gpt2-verifycode-caption")
-feature_extractor = ViTImageProcessor.from_pretrained("TheKOG/vit-gpt2-verifycode-caption")
-tokenizer = AutoTokenizer.from_pretrained("TheKOG/vit-gpt2-verifycode-caption")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)

 import torch
 from PIL import Image
+model = VisionEncoderDecoderModel.from_pretrained("AIris-Channel/vit-gpt2-verifycode-caption")
+feature_extractor = ViTImageProcessor.from_pretrained("AIris-Channel/vit-gpt2-verifycode-caption")
+tokenizer = AutoTokenizer.from_pretrained("AIris-Channel/vit-gpt2-verifycode-caption")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)