Spaces:

Daniton
/

Captioning

Sleeping

Daniton commited on Mar 2, 2023

Commit

2b26d59

1 Parent(s): f1d5627

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model, _, transform = open_clip.create_model_and_transforms(
     "coca_ViT-B-32",
-    pretrained="mscoco_finetuned_CoCa-ViT-B-32-laion2B-s13B-b90k"
 )
 model.to(device)

 model, _, transform = open_clip.create_model_and_transforms(
     "coca_ViT-B-32",
+    pretrained="laion2b_s13b_b90k"
 )
 model.to(device)