Update to medium-sized model

app.py

@@ -9,8 +9,8 @@ import requests
 from PIL import Image
 import matplotlib.pyplot as plt
 
-feature_extractor = ImageGPTFeatureExtractor.from_pretrained("openai/imagegpt-small")
-model = ImageGPTForCausalLM.from_pretrained("openai/imagegpt-small")
+feature_extractor = ImageGPTFeatureExtractor.from_pretrained("openai/imagegpt-medium")
+model = ImageGPTForCausalLM.from_pretrained("openai/imagegpt-medium")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)