spuun commited on
Commit
0de4b9b
·
1 Parent(s): 982b1d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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='base')
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