mustafa2ak commited on
Commit
e715d4b
·
verified ·
1 Parent(s): be4c319

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +5 -6
config.py CHANGED
@@ -3,16 +3,15 @@
3
  # --- Hugging Face Repository Details ---
4
  # The repo where your dataset, images, and trained model are stored.
5
  REPO_ID = "mustafa2ak/Flickr8k-Images"
6
- IMAGE_DIR_NAME = "Flicker8k_Dataset" # The folder with the original images
7
 
8
  # --- File Names ---
9
- # The names of the artifact files you uploaded.
10
  SAVED_MODEL_PATH = "flickr8k_clip_model.pth"
11
  IMAGE_EMBEDDINGS_PATH = "image_embeddings.pt"
12
  IMAGE_PATHS_PKL = "image_paths.pkl"
13
 
14
  # --- Model Dimensions ---
15
- # These must match the encoders you used during training.
16
- IMAGE_EMBEDDING_DIM = 2048 # From ResNet50
17
- TEXT_EMBEDDING_DIM = 768 # From DistilBERT
18
- PROJECTION_DIM = 256 # The size of the shared space
 
3
  # --- Hugging Face Repository Details ---
4
  # The repo where your dataset, images, and trained model are stored.
5
  REPO_ID = "mustafa2ak/Flickr8k-Images"
6
+ IMAGE_DIR_NAME = "Flicker8k_Dataset"
7
 
8
  # --- File Names ---
9
+ # The names of the artifact files in your dataset repo.
10
  SAVED_MODEL_PATH = "flickr8k_clip_model.pth"
11
  IMAGE_EMBEDDINGS_PATH = "image_embeddings.pt"
12
  IMAGE_PATHS_PKL = "image_paths.pkl"
13
 
14
  # --- Model Dimensions ---
15
+ IMAGE_EMBEDDING_DIM = 2048
16
+ TEXT_EMBEDDING_DIM = 768
17
+ PROJECTION_DIM = 256