Update app.py

app.py

@@ -85,9 +85,9 @@ 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 = open_clip.create_model_and_transforms('ViT-B/32', pretrained='openai', device=device)
-image_size = model_clip.visual.image_size
-preprocess_clip = open_clip.image_transform('ViT-B/32', is_train=False)
+model_clip = open_clip.create_model('ViT-B/32', pretrained='openai').to(device)
+image_size = (224, 224)
+preprocess_clip = open_clip.image_transform(image_size=image_size, is_train=False)
 tokenizer_clip = open_clip.get_tokenizer('ViT-B/32')
 model_clip.eval()