Update README.md

README.md

@@ -61,9 +61,20 @@ state_dict_path = hf_hub_download(repo_id="lschmidt/edsr-dsc", filename="pytorch
 state_dict = torch.load(state_dict_path, map_location="cpu")
 model.load_state_dict(state_dict, strict=False)
 
-# sample input: must be a 4D tensor (B, C=2, H, W)
+# create random input: must be a 4D tensor (B, C=2, H, W)
 inputs = torch.randn(1, 2, 64, 64)  # replace with coarse wind velocity fields
 
+# or use sample data
+import xarray as xr
+import numpy as np
+
+data_path = hf_hub_download(repo_id="lschmidt/edsr-dsc/test_data", filename="test_wind_velocities.nc")
+ds = xr.open_dataset(data_path)
+u = ds["u100"].values[0]
+v = ds["v100"].values[0]
+inputs = torch.from_numpy(np.stack([u, v], axis=0)).unsqueeze(0).float()  # shape (1, 2, H, W)
+
+
 # prediction
 outputs = model(inputs)