Spaces:

nielsr
/

imagegpt-completion

Runtime error

nielsr HF staff commited on Nov 29, 2021

Commit

09e8220

•

1 Parent(s): 7283bfa

Rename ImageGPT

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import os
 os.system('pip install git+https://github.com/huggingface/transformers --upgrade')
 import gradio as gr
-from transformers import ImageGPTFeatureExtractor, ImageGPTForCausalLM
 import torch
 import numpy as np
 import requests
@@ -10,7 +10,7 @@ from PIL import Image
 import matplotlib.pyplot as plt
 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)

 os.system('pip install git+https://github.com/huggingface/transformers --upgrade')
 import gradio as gr
+from transformers import ImageGPTFeatureExtractor, ImageGPTForCausalImageModeling
 import torch
 import numpy as np
 import requests
 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)