config.py

import os
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 = {
    "amazon": "LofiAmazon/BOLD-Embeddings-Ecolayers-Amazon",
}

HUGGINGFACE_DW_URL = "https://huggingface.co/datasets/LofiAmazon/Global-Ecolayers/resolve/main/{filename}?download=true"

LAYER_NAMES = [
    "median_elevation_1km.tiff",
    "human_footprint.tiff",
    "population_density_1km.tif",
    "annual_precipitation.tif",
    "precipitation_seasonality.tif",
    "annual_mean_air_temp.tif",
    "temp_seasonality.tif",
]