Martijn van Beers commited on
Commit
0c9f8df
1 Parent(s): 5f8002c

Add model urls

Browse files

Add urls for the new ViT/L models

Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -20,6 +20,8 @@ from CLIP_explainability.utils import interpret, show_img_heatmap, show_heatmap_
20
  clip.clip._MODELS = {
21
  "ViT-B/32": "https://openaipublic.azureedge.net/clip/models/40d365715913c9da98579312b702a82c18be219cc2a73407c4526f58eba950af/ViT-B-32.pt",
22
  "ViT-B/16": "https://openaipublic.azureedge.net/clip/models/5806e77cd80f8b59890b7e101eabd078d9fb84e6937f9e85e4ecb61988df416f/ViT-B-16.pt",
 
 
23
  }
24
 
25
  colour_map = {
 
20
  clip.clip._MODELS = {
21
  "ViT-B/32": "https://openaipublic.azureedge.net/clip/models/40d365715913c9da98579312b702a82c18be219cc2a73407c4526f58eba950af/ViT-B-32.pt",
22
  "ViT-B/16": "https://openaipublic.azureedge.net/clip/models/5806e77cd80f8b59890b7e101eabd078d9fb84e6937f9e85e4ecb61988df416f/ViT-B-16.pt",
23
+ "ViT-L/14": "https://openaipublic.azureedge.net/clip/models/b8cca3fd41ae0c99ba7e8951adf17d267cdb84cd88be6f7c2e0eca1737a03836/ViT-L-14.pt",
24
+ "ViT-L/14@336px": "https://openaipublic.azureedge.net/clip/models/3035c92b350959924f9f00213499208652fc7ea050643e8b385c2dac08641f02/ViT-L-14-336px.pt",
25
  }
26
 
27
  colour_map = {