Update app.py
Browse files
app.py
CHANGED
@@ -21,7 +21,7 @@ transform_vq = transforms.Compose([
|
|
21 |
|
22 |
model_url_vq = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/models/model*_vqa.pth'
|
23 |
|
24 |
-
model_vq = blip_vqa(pretrained=model_url_vq, image_size=480, vit='
|
25 |
model_vq.eval()
|
26 |
model_vq = model_vq.to(device)
|
27 |
|
|
|
21 |
|
22 |
model_url_vq = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/models/model*_vqa.pth'
|
23 |
|
24 |
+
model_vq = blip_vqa(pretrained=model_url_vq, image_size=480, vit='large')
|
25 |
model_vq.eval()
|
26 |
model_vq = model_vq.to(device)
|
27 |
|