Spaces:
Runtime error
Runtime error
Update config.py
Browse files
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"
|
| 7 |
|
| 8 |
# --- File Names ---
|
| 9 |
-
# The names of the artifact files
|
| 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 |
-
|
| 16 |
-
|
| 17 |
-
|
| 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
|
|
|