Update README.md

README.md

@@ -123,11 +123,12 @@ map_scatter(data_v021_2t_xr.latitude, data_v021_2t_xr.longitude, data_v021_2t_xr
             cmap_label="Temperature (K)"
             )
 
+```
 <div style="display: flex; justify-content: center;">
-  <img src="2t_aifs_v021.jpeg" alt="Encoder graph" style="width: 50%;"/>
+  <img src="2t_aifs_v021.png" alt="AIFS 2m Temperature" style="width: 50%;"/>
 </div>
 
-```
+
 
 ## Training Details
 
@@ -139,7 +140,7 @@ AIFS is trained to produce 6-hour forecasts. It receives as input a representati
 at \\(t_{−6h}\\), \\(t_{0}\\), and then forecasts the state at time \\(t_{+6h}\\). 
 
 <div style="display: flex; justify-content: center;">
-  <img src="aifs_diagram.png" alt="High-level AIFS diagram" style="width: 80%;"/>
+  <img src="aifs_diagram.png" alt="AIFS 2m Temperature" style="width: 80%;"/>
 </div>
 
 The full list of input and output fields is shown below: