Download ecolayers from HF dataset

app.py

@@ -12,13 +12,22 @@ from sklearn.preprocessing import LabelEncoder
 import torch
 from torch import nn
 from transformers import BertConfig, BertForMaskedLM, PreTrainedTokenizerFast
-from huggingface_hub import PyTorchModelHubMixin
+from huggingface_hub import PyTorchModelHubMixin, hf_hub_download
 from pinecone import Pinecone
 import rasterio
 from rasterio.sample import sample_gen
 
 from config import DEFAULT_INPUTS, MODELS, DATASETS, ID_TO_GENUS_MAP, LAYER_NAMES
 
+# Download ecolayers from HF dataset
+for image_name in LAYER_NAMES:
+    hf_hub_download(
+        repo_id="LofiAmazon/Global-Ecolayers",
+        filename=image_name,
+        repo_type="dataset",
+        local_dir=".",
+    )
+
 # We need this for the eco layers because they are too big
 PIL.Image.MAX_IMAGE_PIXELS = None