changes

imageCaptionGPT.py

@@ -1,5 +1,5 @@
 import pickle, os, io, re, gc
-from transformers import AutoTokenizer, ViTFeatureExtractor, VisionEncoderDecoderModel
+from transformers import AutoTokenizer, VisionEncoderDecoderModel, ViTImageProcessor
 from keras.models import Model, load_model
 from keras.applications.inception_v3 import InceptionV3, preprocess_input
 import numpy as np
@@ -40,7 +40,7 @@ device='cpu'
 encoder_checkpoint = "nlpconnect/vit-gpt2-image-captioning"
 decoder_checkpoint = "nlpconnect/vit-gpt2-image-captioning"
 model_checkpoint = "nlpconnect/vit-gpt2-image-captioning"
-feature_extractor = ViTFeatureExtractor.from_pretrained(encoder_checkpoint, cache_dir=hf_model)
+feature_extractor = ViTImageProcessor.from_pretrained(encoder_checkpoint, cache_dir=hf_model)
 tokenizer = AutoTokenizer.from_pretrained(decoder_checkpoint, cache_dir=hf_model)
 model = VisionEncoderDecoderModel.from_pretrained(model_checkpoint, cache_dir=hf_model)