CLIP_prefix_captioning

Sleeping

ahmedmbutt commited on May 4

Commit

2ecaf41

•

1 Parent(s): 68e0dca

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -227,7 +227,7 @@ def generate2(
 is_gpu = False
 device = CUDA(0) if is_gpu else "cpu"
 clip_model, preprocess = clip.load("ViT-B/32", device=device, jit=False)
-tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
 def inference(img,model_name):
   prefix_length = 10

 is_gpu = False
 device = CUDA(0) if is_gpu else "cpu"
 clip_model, preprocess = clip.load("ViT-B/32", device=device, jit=False)
+tokenizer = GPT2Tokenizer.from_pretrained("gpt2", revision="6c0e6080953db56375760c0471a8c5f2929baf11")
 def inference(img,model_name):
   prefix_length = 10