src/constants.py

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

import os
import pathlib


DATASET_NAME = "Xmaster6y/fruit-vegetable-concepts"
CONCEPTS = [
    # Shapes
    "sphere",
    "cube",
    "cylinder",
    # Colors
    "red",
    "green",
    "orange",
    "yellow",
]
ASSETS_FOLDER = pathlib.Path(__file__).parent / "assets"

HF_TOKEN = os.getenv("HF_TOKEN")