Update README.md

README.md

@@ -31,7 +31,8 @@ The model is intended to be used for any analysis where the reputation (such as
 You can use these models in your own applications by leveraging the Hugging Face Transformers library. Below is a Python code snippet demonstrating how to load and use the FT-News classification model:
 
 ```python
-from transformers import pipeline
+from peft import AutoPeftModelForCausalLM
+from transformers import AutoTokenizer, pipeline
 
 # Load the model
 model = AutoPeftModelForCausalLM.from_pretrained("Moritz-Pfeifer/financial-times-classification-llama-2-7b-v1.3")