Improve model card: Add pipeline tag and library name

README.md

@@ -1,9 +1,13 @@
 ---
 license: apache-2.0
+library_name: terratorch
+pipeline_tag: image-feature-extraction
 ---
 
 #  Model Card for granite-geospatial-biomass
 
+This model is based on the paper [TerraTorch: The Geospatial Foundation Models Toolkit](https://huggingface.co/papers/2503.20563).
+
 <p align="center" width="100%">
 <img src="https://github.com/ibm-granite/granite-geospatial-biomass/blob/main/biomass-image.jpeg?raw=true" width="600">
 </p>
@@ -25,6 +29,7 @@ You can use it easily with Terratorch through:
 
 ```python
 from terratorch.cli_tools import LightningInferenceModel
+from huggingface_hub import hf_hub_download
 
 ckpt_path = hf_hub_download(repo_id="ibm-granite/granite-geospatial-biomass", filename="biomass_model.ckpt")
 config_path = hf_hub_download(repo_id="ibm-granite/granite-geospatial-biomass", filename="config.yaml")