ktllc commited on
Commit
c812dab
·
1 Parent(s): 3882946

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ model_type = "vit_h"
14
  sam = sam_model_registry[model_type](checkpoint=sam_checkpoint)
15
 
16
  # Load the CLIP model
17
- model, preprocess = clip.load("ViT-B/32")
18
  device = "cuda" if torch.cuda.is_available() else "cpu"
19
  model.to(device).eval()
20
 
 
14
  sam = sam_model_registry[model_type](checkpoint=sam_checkpoint)
15
 
16
  # Load the CLIP model
17
+ model, preprocess = clip.load("ViT-L/14@336px")
18
  device = "cuda" if torch.cuda.is_available() else "cpu"
19
  model.to(device).eval()
20