MusIre commited on
Commit
7e30be4
·
verified ·
1 Parent(s): 1518bac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -86,7 +86,7 @@ scheduler = ReduceLROnPlateau(optimizer, mode='min', factor=0.5, patience=3, ver
86
 
87
  # Load GPT-Neo and CLIP
88
  model_clip = open_clip.create_model('ViT-B/32', pretrained='openai').to(device)
89
- preprocess_clip = open_clip.image_transform('ViT-B/32')
90
  tokenizer_clip = open_clip.get_tokenizer('ViT-B/32')
91
  model_clip.eval()
92
 
 
86
 
87
  # Load GPT-Neo and CLIP
88
  model_clip = open_clip.create_model('ViT-B/32', pretrained='openai').to(device)
89
+ preprocess_clip = open_clip.image_transform('ViT-B/32', is_train=False)
90
  tokenizer_clip = open_clip.get_tokenizer('ViT-B/32')
91
  model_clip.eval()
92