Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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-
|
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 |
|