Update README.md

README.md

@@ -36,8 +36,8 @@ be fine-tuned on a downstream task.
 Here is how to initialize this model in PyTorch:
 
 ```python
-from transformers import BartModel
-model = BartModel.from_pretrained("xfbai/AMRBART-base")
+from transformers import BartForConditionalGeneration
+model = BartForConditionalGeneration.from_pretrained("xfbai/AMRBART-base")
 ```
 Please refer to [this repository](https://github.com/muyeby/AMRBART) for tokenizer initialization and data preprocessing.