Create imagegpt

imagegpt

@@ -0,0 +1,13 @@
+from transformers import AutoImageProcessor, ImageGPTModel
+from PIL import Image
+import requests
+
+url = "http://images.cocodataset.org/val2017/000000039769.jpg"
+image = Image.open(requests.get(url, stream=True).raw)
+
+image_processor = AutoImageProcessor.from_pretrained("openai/imagegpt-small")
+model = ImageGPTModel.from_pretrained("openai/imagegpt-small")
+
+inputs = image_processor(images=image, return_tensors="pt")
+outputs = model(**inputs)
+last_hidden_states = outputs.last_hidden_state