Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -161,7 +161,6 @@ change_requires_grad(image_encoder, False)
|
|
161 |
tokenizer, model = load_checkpoint(ckpt_path, load_from_epoch)
|
162 |
model.eval()
|
163 |
|
164 |
-
|
165 |
img_transform = transforms.Compose([transforms.Resize((img_size,img_size)),
|
166 |
transforms.ToTensor(),
|
167 |
transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])])
|
|
|
161 |
tokenizer, model = load_checkpoint(ckpt_path, load_from_epoch)
|
162 |
model.eval()
|
163 |
|
|
|
164 |
img_transform = transforms.Compose([transforms.Resize((img_size,img_size)),
|
165 |
transforms.ToTensor(),
|
166 |
transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])])
|