ktllc commited on
Commit
feb267b
·
1 Parent(s): 5cbcd45

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from io import BytesIO
8
  from decimal import Decimal
9
 
10
  # Load the CLIP model
11
- model, preprocess = clip.load("ViT-B/32")
12
  device = "cuda" if torch.cuda.is_available() else "cpu"
13
  model.to(device).eval()
14
 
 
8
  from decimal import Decimal
9
 
10
  # Load the CLIP model
11
+ model, preprocess = clip.load("ViT-L/14@336px")
12
  device = "cuda" if torch.cuda.is_available() else "cpu"
13
  model.to(device).eval()
14