Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -24,7 +24,7 @@ clip_device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 24 |
|
| 25 |
# Create a writable models directory
|
| 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="./models")
|
| 30 |
|
|
|
|
| 24 |
|
| 25 |
# Create a writable models directory
|
| 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="./models")
|
| 30 |
|