Spaces:
Sleeping
Sleeping
use gh model
Browse files
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="
|
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)
|