imthanhlv commited on
Commit
547178e
1 Parent(s): 93007bd

fixed tokenizr

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -224,7 +224,8 @@ def generate2(
224
  is_gpu = False
225
  device = CUDA(0) if is_gpu else "cpu"
226
  clip_model, preprocess = clip.load("ViT-B/16", device=device, jit=False)
227
- tokenizer = GPT2Tokenizer.from_pretrained("imthanhlv/gpt2news")
 
228
 
229
  def inference(img, text, is_translate):
230
  prefix_length = 10
 
224
  is_gpu = False
225
  device = CUDA(0) if is_gpu else "cpu"
226
  clip_model, preprocess = clip.load("ViT-B/16", device=device, jit=False)
227
+ from transformers import AutoTokenizer
228
+ tokenizer = AutoTokenizer.from_pretrained("imthanhlv/gpt2news")
229
 
230
  def inference(img, text, is_translate):
231
  prefix_length = 10