Update app.py

app.py

@@ -4,6 +4,7 @@ from PIL import Image
 from torchvision import transforms, models
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import pandas as pd
+import open_clip
 import random
 import urllib.parse
 import torch.nn as nn
@@ -85,7 +86,8 @@ optimizer = torch.optim.Adam(model_resnet.parameters(), lr=0.001, weight_decay=1
 scheduler = ReduceLROnPlateau(optimizer, mode='min', factor=0.5, patience=3, verbose=True)
 
 # Load GPT-Neo and CLIP
-model_clip, preprocess_clip = clip.load("ViT-B/32", device=device)
+model_clip, preprocess_clip = open_clip.create_model_and_transforms('ViT-B/32', device=device)
+tokenizer_clip = open_clip.get_tokenizer('ViT-B/32')
 model_clip.eval()
 
 model_name = "EleutherAI/gpt-neo-1.3B"