Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -26,7 +26,7 @@ clip_device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 26 |
# os.makedirs("models", exist_ok=True)
|
| 27 |
os.makedirs("/tmp/models", exist_ok=True)
|
| 28 |
|
| 29 |
-
clip_model, clip_preprocess = clip.load("ViT-B/32", device=clip_device, download_root="
|
| 30 |
|
| 31 |
|
| 32 |
# Class names
|
|
|
|
| 26 |
# os.makedirs("models", exist_ok=True)
|
| 27 |
os.makedirs("/tmp/models", exist_ok=True)
|
| 28 |
|
| 29 |
+
clip_model, clip_preprocess = clip.load("ViT-B/32", device=clip_device, download_root="/tmp/models")
|
| 30 |
|
| 31 |
|
| 32 |
# Class names
|