Spaces:
Build error
Build error
RobotJelly
commited on
Commit
•
7717b60
1
Parent(s):
c35bf48
app.py
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ from sentence_transformers import SentenceTransformer, util
|
|
15 |
# check if CUDA available
|
16 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
17 |
|
18 |
-
IMAGES_DIR = Path("
|
19 |
|
20 |
#Load CLIP model
|
21 |
model = SentenceTransformer('clip-ViT-B-32')
|
|
|
15 |
# check if CUDA available
|
16 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
17 |
|
18 |
+
IMAGES_DIR = Path("photos/")
|
19 |
|
20 |
#Load CLIP model
|
21 |
model = SentenceTransformer('clip-ViT-B-32')
|