Improve the README.

README.md

@@ -71,13 +71,18 @@ cd BiRefNet
 ```
 
 ```python
-# Load weights
+# Use codes locally
 from models.birefnet import BiRefNet
 
-# Option-1: From Hugging Face Models
+# Load weights from Hugging Face Models
 birefnet = BiRefNet.from_pretrained('zhengpeng7/birefnet')
+```
+
+#### Use codes from GitHub + weights from HuggingFace
+> Only use the weights and codes both locally.
 
-# Option-2: From local disk
+```python
+# Use codes and weights locally
 import torch
 from utils import check_state_dict
 
@@ -85,7 +90,6 @@ 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)
-
 ```
 
 #### Use the loaded BiRefNet for inference