Update README.md

README.md

@@ -1,16 +1,19 @@
 ---
-pipeline_tag: image-to-text
 license: apache-2.0
+pipeline_tag: image-to-text
+tags:
+- image-captioning
+- image-to-text
 ---
-## Usage method:
+## Use in Transformers
 ```python
 from transformers import VisionEncoderDecoderModel, ViTImageProcessor, AutoTokenizer
 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")
+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)