Update README.md

README.md

@@ -52,6 +52,15 @@ Visit our GitHub repo: [https://github.com/ZhengPeng7/BiRefNet](https://github.c
 
 ### 1. Load BiRefNet:
 
+#### Use codes + weights from HuggingFace
+> Only use the weights on HuggingFace -- Pro: No need to download BiRefNet codes manually; Con: Codes on HuggingFace might not be latest version (I'll try to keep them always latest).
+
+```python
+# Load BiRefNet with weights
+from transformers import AutoModelForImageSegmentation
+birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/birefnet', trust_remote_code=True)
+```
+
 #### Use codes from GitHub + weights from HuggingFace
 > Only use the weights on HuggingFace -- Pro: codes are always latest; Con: Need to clone the BiRefNet repo from my GitHub.
 
@@ -64,16 +73,19 @@ cd BiRefNet
 ```python
 # Load weights
 from models.birefnet import BiRefNet
+
+# Option-1: From Hugging Face Models
 birefnet = BiRefNet.from_pretrained('zhengpeng7/birefnet')
-```
 
-#### Use codes + weights from HuggingFace
-> Only use the weights on HuggingFace -- Pro: No need to download BiRefNet codes manually; Con: Codes on HuggingFace might not be latest version (I'll try to keep them always latest).
+# Option-2: From local disk
+import torch
+from utils import check_state_dict
+
+birefnet = BiRefNet(bb_pretrained=False)
+state_dict = torch.load(PATH_TO_WEIGHT, map_location='cpu')
+state_dict = check_state_dict(state_dict)
+birefnet.load_state_dict(state_dict)
 
-```python
-# Load BiRefNet with weights
-from transformers import AutoModelForImageSegmentation
-birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/birefnet', trust_remote_code=True)
 ```
 
 #### Use the loaded BiRefNet for inference