Update readme

README.md

@@ -41,12 +41,12 @@ torch.set_default_device('cpu')  # or 'cuda'
 
 # create model
 model = AutoModelForCausalLM.from_pretrained(
-    'BAAI/bunny-phi-2-siglip',
+    'BAAI/Bunny-v1_0-3B',
     torch_dtype=torch.float16,
     device_map='auto',
     trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained(
-    'BAAI/bunny-phi-2-siglip',
+    'BAAI/Bunny-v1_0-3B',
     trust_remote_code=True)
 
 # text prompt
@@ -55,7 +55,7 @@ text = f"A chat between a curious user and an artificial intelligence assistant.
 text_chunks = [tokenizer(chunk).input_ids for chunk in text.split('<image>')]
 input_ids = torch.tensor(text_chunks[0] + [-200] + text_chunks[1], dtype=torch.long).unsqueeze(0)
 
-# image, sample images can be found in https://huggingface.co/BAAI/bunny-phi-2-siglip/tree/main/images
+# image, sample images can be found in images folder
 image = Image.open('example_2.png')
 image_tensor = model.process_images([image], model.config).to(dtype=model.dtype)
 
@@ -72,7 +72,7 @@ print(tokenizer.decode(output_ids[input_ids.shape[1]:], skip_special_tokens=True
 Before running the snippet, you need to install the following dependencies:
 
 ```shell
-pip install torch transformers accelerate
+pip install torch transformers accelerate pillow
 ```
 
 # License