Update README.md

README.md

@@ -24,6 +24,9 @@ For install instructions, please see https://github.com/EPFL-VILAB/ViPer.
 This model can be loaded from Hugging Face Hub as follows:
 
 ```python
+from transformers import AutoModelForVision2Seq
+from peft import PeftModel
+
 model = AutoModelForVision2Seq.from_pretrained("HuggingFaceM4/idefics2-8b").to(device)
 model = PeftModel.from_pretrained(model, "EPFL-VILAB/VPE-ViPer").to(device)
 ```