Update clip_chat.py

clip_chat.py

@@ -8,7 +8,7 @@ from random import choice
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model, preprocess = clip.load("ViT-L/14@336px", device=device)
-COCO = glob.glob(os.path.join(os.getcwd(), "train2017", "*"))
+COCO = glob.glob(os.path.join(os.getcwd(), "images", "*"))
 # with open('coco_paths.txt', 'r') as file:
 #     COCO = file.read().split('\n')[:-1]
 # COCO = [c.replace("/media/bonilla/My Book/coco/train2017/", "D:\\coco\\train2017\\") for c in COCO]