nielsr HF Staff commited on
Commit
a4be8a3
·
verified ·
1 Parent(s): 32a02e8

Improve model card: Add pipeline tag and library name

Browse files

This PR improves the model card by adding the `image-feature-extraction` pipeline tag and specifying `terratorch` as the library name. This enhances discoverability and provides more context for users. The PR also links to the paper on which the model is based at the top of the model card. It should now be easier for users to locate and evaluate the model's performance. It also links to the base model's paper, ensuring people can find it at https://huggingface.co/papers/2310.18660.

Files changed (1) hide show
  1. README.md +5 -0
README.md CHANGED
@@ -1,9 +1,13 @@
1
  ---
2
  license: apache-2.0
 
 
3
  ---
4
 
5
  # Model Card for granite-geospatial-biomass
6
 
 
 
7
  <p align="center" width="100%">
8
  <img src="https://github.com/ibm-granite/granite-geospatial-biomass/blob/main/biomass-image.jpeg?raw=true" width="600">
9
  </p>
@@ -25,6 +29,7 @@ You can use it easily with Terratorch through:
25
 
26
  ```python
27
  from terratorch.cli_tools import LightningInferenceModel
 
28
 
29
  ckpt_path = hf_hub_download(repo_id="ibm-granite/granite-geospatial-biomass", filename="biomass_model.ckpt")
30
  config_path = hf_hub_download(repo_id="ibm-granite/granite-geospatial-biomass", filename="config.yaml")
 
1
  ---
2
  license: apache-2.0
3
+ library_name: terratorch
4
+ pipeline_tag: image-feature-extraction
5
  ---
6
 
7
  # Model Card for granite-geospatial-biomass
8
 
9
+ This model is based on the paper [TerraTorch: The Geospatial Foundation Models Toolkit](https://huggingface.co/papers/2503.20563).
10
+
11
  <p align="center" width="100%">
12
  <img src="https://github.com/ibm-granite/granite-geospatial-biomass/blob/main/biomass-image.jpeg?raw=true" width="600">
13
  </p>
 
29
 
30
  ```python
31
  from terratorch.cli_tools import LightningInferenceModel
32
+ from huggingface_hub import hf_hub_download
33
 
34
  ckpt_path = hf_hub_download(repo_id="ibm-granite/granite-geospatial-biomass", filename="biomass_model.ckpt")
35
  config_path = hf_hub_download(repo_id="ibm-granite/granite-geospatial-biomass", filename="config.yaml")