RobotJelly commited on
Commit
7717b60
1 Parent(s): c35bf48
Files changed (1) hide show
  1. app.py +1 -1
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("./photos/")
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')