Update config.py

config.py

@@ -4,12 +4,19 @@ 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)
 
-HF_DATASET_REPO = os.environ.get("HF_DATASET_REPO", "CircleStar/charcoal-microscopy")
+# Replace this with your real private dataset repo
+HF_DATASET_REPO = os.environ.get("HF_DATASET_REPO", "YOUR_USERNAME/charcoal-microscopy")
+
+# Must be added in Hugging Face Space Settings → Secrets
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
 IMAGE_SIZE = 224
-RANDOM_SEED = 42
+RANDOM_SEED = 42
+
+DATASET_DISPLAY_NAME = "Images microscopiques de charbons de bois"