ahmedmbutt
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -227,7 +227,7 @@ def generate2(
|
|
227 |
is_gpu = False
|
228 |
device = CUDA(0) if is_gpu else "cpu"
|
229 |
clip_model, preprocess = clip.load("ViT-B/32", device=device, jit=False)
|
230 |
-
tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
|
231 |
|
232 |
def inference(img,model_name):
|
233 |
prefix_length = 10
|
|
|
227 |
is_gpu = False
|
228 |
device = CUDA(0) if is_gpu else "cpu"
|
229 |
clip_model, preprocess = clip.load("ViT-B/32", device=device, jit=False)
|
230 |
+
tokenizer = GPT2Tokenizer.from_pretrained("gpt2", revision="6c0e6080953db56375760c0471a8c5f2929baf11")
|
231 |
|
232 |
def inference(img,model_name):
|
233 |
prefix_length = 10
|