Spaces:
Running
Running
File size: 583 Bytes
7a4d9b4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from pathlib import Path
import os
KEEP_KEYS = {
# ==global UI context we want to keep after "Reset"==
"model_select", # sidebar model key
"input_mode", # radio for Upload|Sample
"uploader_version", # version counter for file uploader
"input_registry", # radio controlling Upload vs Sample
}
TARGET_LEN = 500
SAMPLE_DATA_DIR = Path("sample_data")
MODEL_WEIGHTS_DIR = os.getenv("WEIGHTS_DIR") or (
"model_weights" if os.path.isdir("model_weights") else "outputs"
)
# ==Label mapping==
LABEL_MAP = {0: "Stable (Unweathered)", 1: "Weathered (Degraded)"}
|