gpucce commited on
Commit
21a0d95
1 Parent(s): 5270787

use gh model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ device = torch.device("cuda")
6
 
7
  model, _, transform = open_clip.create_model_and_transforms(
8
  "coca_ViT-L-14",
9
- pretrained="laion2B-s13B-b90k-mscoco-2014.pt"
10
  )
11
 
12
  model.to(device)
 
6
 
7
  model, _, transform = open_clip.create_model_and_transforms(
8
  "coca_ViT-L-14",
9
+ pretrained="mscoco_finetuned_laion2B-s13B-b90k"
10
  )
11
 
12
  model.to(device)