src/constants.py

"""Constants use in the space.
"""

import os
import pathlib


DATASET_NAME = "mulsi/fruit-vegetable-concepts"
CONCEPTS = [
    # Environment
    "stem",
    "leaf",
    "tail",
    "seed",
    "pulp",
    "soil",
    "tree",
    # Shapes
    "ovaloid",
    "sphere",
    "cylinder",
    "cube",
    # Colors
    "black",
    "purple",
    "red",
    "blue",
    "green",
    "brown",
    "orange",
    "yellow",
    "white",
]
ASSETS_FOLDER = pathlib.Path(__file__).parent / "assets"

HF_TOKEN = os.getenv("HF_TOKEN")