Update README.md

README.md

@@ -31,6 +31,7 @@ This is the 4-bit version of InternLM-XComposer2, install the latest version of
 
 ```python
 import torch, auto_gptq
+from PIL import Image
 from transformers import AutoModel, AutoTokenizer 
 from auto_gptq.modeling import BaseGPTQForCausalLM
 
@@ -62,7 +63,7 @@ img_path_list = [
 images = []
 for img_path in img_path_list:
     image = Image.open(img_path).convert("RGB")
-    image = quant_model.vis_processor(image)
+    image = model.vis_processor(image)
     images.append(image)
 image = torch.stack(images)
 query = '<ImageHere> <ImageHere>please write an article based on the images. Title: my favorite animal.'