config.py

# config.py

# --- Hugging Face Repository Details ---
# The repo where your dataset, images, and trained model are stored.
REPO_ID = "mustafa2ak/Flickr8k-Images"
IMAGE_DIR_NAME = "Flicker8k_Dataset"

# --- File Names ---
# The names of the artifact files in your dataset repo.
SAVED_MODEL_PATH = "flickr8k_clip_model.pth"
IMAGE_EMBEDDINGS_PATH = "image_embeddings.pt"
IMAGE_PATHS_PKL = "image_paths.pkl"

# --- Model Dimensions ---
IMAGE_EMBEDDING_DIM = 2048
TEXT_EMBEDDING_DIM = 768
PROJECTION_DIM = 256