Update README.md

README.md

@@ -46,5 +46,16 @@ model.load_state_dict(state_dict, strict=False)
 # generate sample data (B, C, W, H) 
 inputs = torch.randn(1, 2, 10, 10)
 
+# or use test data
+data_path = hf_hub_download(
+    repo_id="lschmidt/rcan-dsc",
+    filename="test_wind_velocities.nc",
+    subfolder="test_data"  
+)
+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()  
+
 # prediction
 output = model(inputs)