Update README.md

README.md

@@ -31,12 +31,12 @@ You can use the raw model for either feature extractor or (un) conditional image
 Here is how to use this model in PyTorch to perform unconditional image generation:
 
 ```python
-from transformers import ImageGPTFeatureExtractor, ImageGPTForCausalImageModeling
+from transformers import ImageGPTImageProcessor, ImageGPTForCausalImageModeling
 import torch
 import matplotlib.pyplot as plt
 import numpy as np
 
-feature_extractor = ImageGPTFeatureExtractor.from_pretrained('openai/imagegpt-small')
+feature_extractor = ImageGPTImageProcessor.from_pretrained('openai/imagegpt-small')
 model = ImageGPTForCausalImageModeling.from_pretrained('openai/imagegpt-small')
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")