CircleStar's picture
Update config.py
c47b92a verified
raw
history blame contribute delete
678 Bytes
import os
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
MODEL_DIR = os.path.join(BASE_DIR, "saved_models")
META_DIR = os.path.join(BASE_DIR, "saved_models_meta")
FIGURE_DIR = os.path.join(BASE_DIR, "saved_figures")
os.makedirs(MODEL_DIR, exist_ok=True)
os.makedirs(META_DIR, exist_ok=True)
os.makedirs(FIGURE_DIR, exist_ok=True)
# Replace this with your real private dataset repo
HF_DATASET_REPO = os.environ.get("HF_DATASET_REPO", "CircleStar/charcoal-microscopy")
# Must be added in Hugging Face Space Settings → Secrets
HF_TOKEN = os.environ.get("HF_TOKEN")
IMAGE_SIZE = 224
RANDOM_SEED = 42
DATASET_DISPLAY_NAME = "Images microscopiques de charbons de bois"