Spaces:
Build error
Build error
JunnanLi
commited on
Commit
•
ea784ed
1
Parent(s):
4214d1d
Update app.py
Browse files
app.py
CHANGED
@@ -21,7 +21,7 @@ transform = transforms.Compose([
|
|
21 |
transforms.Normalize((0.48145466, 0.4578275, 0.40821073), (0.26862954, 0.26130258, 0.27577711))
|
22 |
])
|
23 |
|
24 |
-
model_url = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/models/
|
25 |
|
26 |
model = blip_decoder(pretrained=model_url, image_size=384, vit='large')
|
27 |
model.eval()
|
|
|
21 |
transforms.Normalize((0.48145466, 0.4578275, 0.40821073), (0.26862954, 0.26130258, 0.27577711))
|
22 |
])
|
23 |
|
24 |
+
model_url = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/models/model_large_caption.pth'
|
25 |
|
26 |
model = blip_decoder(pretrained=model_url, image_size=384, vit='large')
|
27 |
model.eval()
|