Update README.md

README.md

@@ -44,11 +44,22 @@ It is adapted for super-resolution of **2-channel weather data** (e.g., wind u a
 ## 🚀 How to Use
 
 ```python
-from super_image import EdsrModel
+from super_image import EdsrModel, EdsrConfig
+from huggingface_hub import hf_hub_download
 import torch
 
-# Load model and weights directly from Hugging Face Hub
-model = EdsrModel.from_pretrained("lschmidt/edsr-dsc", scale=4)
+# load config 
+config, _ = EdsrConfig.from_pretrained("lschmidt/edsr-dsc")
+
+# load & modify model
+model = EdsrModel(config)
+del model.sub_mean
+del model.add_mean
+
+# load pre-trained weights
+state_dict_path = hf_hub_download(repo_id="lschmidt/edsr-dsc", filename="pytorch_model_4x.pt")
+state_dict = torch.load(state_dict_path, map_location="cpu")
+model.load_state_dict(state_dict, strict=False)
 
 # Prepare input: must be a 4D tensor (B, C=2, H, W)
 inputs = torch.randn(1, 2, 64, 64)  # Replace with actual wind field data