Spaces:
Sleeping
Sleeping
Update config.py
Browse files
config.py
CHANGED
|
@@ -8,7 +8,7 @@ META_DIR = os.path.join(BASE_DIR, "saved_models_meta")
|
|
| 8 |
os.makedirs(MODEL_DIR, exist_ok=True)
|
| 9 |
os.makedirs(META_DIR, exist_ok=True)
|
| 10 |
|
| 11 |
-
HF_DATASET_REPO = os.environ.get("HF_DATASET_REPO", "
|
| 12 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
| 13 |
|
| 14 |
IMAGE_SIZE = 64
|
|
|
|
| 8 |
os.makedirs(MODEL_DIR, exist_ok=True)
|
| 9 |
os.makedirs(META_DIR, exist_ok=True)
|
| 10 |
|
| 11 |
+
HF_DATASET_REPO = os.environ.get("HF_DATASET_REPO", "CircleStar/charcoal-microscopy")
|
| 12 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
| 13 |
|
| 14 |
IMAGE_SIZE = 64
|