Update README.md

README.md

@@ -16,7 +16,7 @@ model developed by the European Centre for Medium-Range Weather Forecasts (ECMWF
 
 We show that AIFS produces highly skilled forecasts for upper-air variables, surface weather parameters and
 tropical cyclone tracks. AIFS is run four times daily alongside ECMWF’s physics-based NWP model and forecasts
-are available to the public under ECMWF’s open data policy.
+are available to the public under ECMWF’s open data policy. (https://www.ecmwf.int/en/forecasts/datasets/open-data)
 
 ## Model Details
 
@@ -46,13 +46,21 @@ and direct observational data.
 
 
 - **Repository:** [Anemoi](https://anemoi-docs.readthedocs.io/en/latest/index.html)
+  Anemoi is an open-source framework for creating machine learning (ML) weather forecasting systems, which ECMWF and a range of national meteorological services across Europe have co-developed.
 - **Paper:** https://arxiv.org/pdf/2406.01465
 
 ## How to Get Started with the Model
 
-Use the code below to get started with the model.
+To be able to run AIFS to generate a new forecast, you can use ai-models https://github.com/ecmwf-lab/ai-models.
+`ai-models` command can be used to run different models, since in this case we are looking at using AIFS we need to speficy
+`anemoi` as `model-name` and then pass the path to the checkpoint (`aifs_single_v0.2.1.ckpt`) and the initial conditions. 
+You can find an example of a set of initial conditions in the GRIB file `example_20241107_12_n320.grib`.
+
+Use the code below to get started with the model. 
+
 
 ```
+# 1st create the conda environment
 export CONDA_ENV=aifs-env
 conda create -n ${CONDA_ENV} python=3.10
 conda activate ${CONDA_ENV}
@@ -62,6 +70,8 @@ pip install anemoi-inference[plugin] anemoi-models==0.2
 pip install ninja
 pip install flash-attn --no-build-isolation
 
+# Run ai-models 
+
 ai-models anemoi --checkpoint aifs_single_v0.2.1.ckpt --file example_20241107_12_n320.grib
 ```
 ! ISSUE WITH PYTORCH 2.5 and flash_attention, and cuda 12.4. For now keep it to torch 2.4