Spaces:

SeemG
/

TextToImageFlickrSearch

Runtime error

SeemG commited on Jul 26, 2024

Commit

e9078d1

verified ·

1 Parent(s): be42ff0

Update clip_model.py

Files changed (1) hide show

clip_model.py CHANGED Viewed

@@ -18,8 +18,8 @@ from transformers import DistilBertModel, DistilBertConfig, DistilBertTokenizer
 class CFG:
     debug = False
-    image_path = "/content/flickr30k_images/flickr30k_images"
-    captions_path = "/content"
     batch_size = 32
     num_workers = 2
     head_lr = 1e-3

 class CFG:
     debug = False
+    image_path = "/flickr30k_images/flickr30k_images"
+    captions_path = "."
     batch_size = 32
     num_workers = 2
     head_lr = 1e-3