Spaces:

twobob
/

imagegpt

Runtime error

twobob commited on Aug 4, 2022

Commit

6cc515e

•

1 Parent(s): 25cc57c

revert medium

Files changed (1) hide show

app.py CHANGED Viewed

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

 from PIL import Image
 import matplotlib.pyplot as plt
+feature_extractor = ImageGPTFeatureExtractor.from_pretrained("openai/imagegpt-medium")
+model = ImageGPTForCausalImageModeling.from_pretrained("openai/imagegpt-medium")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)