| from __future__ import annotations | |
| # DATASET_NAME = "tomytjandra/h-and-m-fashion-caption" | |
| DATASET_NAME = "sagecontinuum/INQUIRE-Benchmark-small" | |
| # Default dataset/view options (can be changed in UI) | |
| DEFAULT_SPLIT = "validation" | |
| SPLIT_CHOICES = ["validation"] | |
| DEFAULT_IMAGE_COL = "image" | |
| DEFAULT_TEXT_COL = "query" | |
| DEFAULT_MAX_SAMPLES = 4000 # cap for demo builds; adjustable in UI | |
| MAX_SAMPLES_CAP_FOR_DEMO = 4000 | |
| DEFAULT_TOP_K = 10 | |
| # Index cache directory inside the Space persistent storage | |
| DEFAULT_INDEX_DIR = "./index_cache" | |
| # SigLIP checkpoints (you can extend this list) | |
| # The first one is set as the default for this app. | |
| CKPT_CHOICES = [ | |
| "vungocthach1112/siglip-inquire-checkpoint", | |
| # "vungocthach1112/siglip-fashion-checkpoint", | |
| "google/siglip-so400m-patch14-384", | |
| ] | |
| DEFAULT_CKPT = CKPT_CHOICES[0] | |
| # FAISS device preference: 'auto' (GPU if available), 'gpu', or 'cpu' | |
| DEFAULT_FAISS_DEVICE = "auto" | |
| # EXAMPLE_QUERIES = [ | |
| # "red floral summer dress", | |
| # "men's black leather jacket", | |
| # "white sneakers with chunky sole", | |
| # "blue denim jeans for women", | |
| # "kids' yellow raincoat", | |
| # ] | |
| EXAMPLE_QUERIES = [ | |
| "leopard on a road", | |
| "red fox in snow", | |
| "hummingbird drinking nectar", | |
| "owl perched in daylight", | |
| "green snake in grass", | |
| ] | |