Create config.py

config.py

@@ -0,0 +1,15 @@
+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")
+
+os.makedirs(MODEL_DIR, exist_ok=True)
+os.makedirs(META_DIR, exist_ok=True)
+
+HF_DATASET_REPO = os.environ.get("HF_DATASET_REPO", "你的用户名/charcoal-microscopy")
+HF_TOKEN = os.environ.get("HF_TOKEN")
+
+IMAGE_SIZE = 64
+RANDOM_SEED = 42