MusIre commited on
Commit
1518bac
·
verified ·
1 Parent(s): 72e4391

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -85,7 +85,8 @@ optimizer = torch.optim.Adam(model_resnet.parameters(), lr=0.001, weight_decay=1
85
  scheduler = ReduceLROnPlateau(optimizer, mode='min', factor=0.5, patience=3, verbose=True)
86
 
87
  # Load GPT-Neo and CLIP
88
- model_clip, preprocess_clip = open_clip.create_model_and_transforms('ViT-B/32', device=device)
 
89
  tokenizer_clip = open_clip.get_tokenizer('ViT-B/32')
90
  model_clip.eval()
91
 
 
85
  scheduler = ReduceLROnPlateau(optimizer, mode='min', factor=0.5, patience=3, verbose=True)
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