Update app.py

app.py

@@ -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", revision="6c0e6080953db56375760c0471a8c5f2929baf11")
+tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
 
 def inference(img,model_name):   
   prefix_length = 10
@@ -238,7 +238,7 @@ def inference(img,model_name):
       model_path = coco_weight
   else:
       model_path = conceptual_weight
-  model.load_state_dict(torch.load(model_path, map_location=CPU)) 
+  model.load_state_dict(torch.load(model_path, map_location=CPU), strict=False) 
   model = model.eval() 
   device = CUDA(0) if is_gpu else "cpu"
   model = model.to(device)