Update README.md

README.md

@@ -24,6 +24,7 @@ We introduce CodeGeeX4-ALL-9B, the open-source version of the latest CodeGeeX4 m
 Use `4.39.0<=transformers<=4.40.2` to quickly launch [codegeex4-all-9b](https://huggingface.co/THUDM/codegeex2-6b)：
 
 ```python
+import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -44,8 +45,7 @@ with torch.no_grad():
 
 ## License
 
-The model weights are licensed under the [Model License](MODEL_LICENSE).
-
+The model weights are licensed under the following [License](./LICENSE).
 
 ## Citation