Update clip_chat.py

clip_chat.py

@@ -7,10 +7,11 @@ 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(), "images", "*"))
 available_models = ['RN50', 'RN101', 'RN50x4', 'RN50x16', 'RN50x64', 'ViT-B/32', 'ViT-B/16', 'ViT-L/14', 'ViT-L/14@336px']
+model, preprocess = clip.load(available_models[-1], device=device)
+
+COCO = glob.glob(os.path.join(os.getcwd(), "images", "*"))
 
 
 def load_random_image():