Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|