config.py

import json


with open("default_inputs.json", "r") as default_inputs_file:
    DEFAULT_INPUTS = json.load(default_inputs_file)

with open("genus_labels.json", "r") as genus_labels_file:
    GENUS_TO_ID_MAP = json.load(genus_labels_file)
    ID_TO_GENUS_MAP = {v: k for k, v in GENUS_TO_ID_MAP.items()}

ECOLAYERS = [
    "median_elevation_1km",
    "human_footprint",
    "population_density_1km",
    "annual_precipitation",
    "precipitation_seasonality",
    "annual_mean_air_temp",
    "temp_seasonality",
]

MODELS = {
    "embeddings": "LofiAmazon/BarcodeBERT-Entire-BOLD",
    "classification": "LofiAmazon/BarcodeBERT-Finetuned-Amazon",
}

DATASETS = {
    "ecolayers": "LofiAmazon/Global-Ecolayers",
}