Update README.md (#2)

- Update README.md (6cf334e9996110e15d3c27857922e3f8b21e5071)

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,12 +63,12 @@ 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.'
 with torch.cuda.amp.autocast():
-    response, history = quant_model.chat(tokenizer, query=query, image=image, history=[], do_sample=False)
+    response, history = model.chat(tokenizer, query=query, image=image, history=[], do_sample=False)
 print(response)
 
 #My Favorite Animal: The Panda