Update README.md

README.md

@@ -166,8 +166,8 @@ messages = [
     },       
 ]
 prompt = processor.apply_chat_template(messages, add_generation_prompt=True)
-inputs = processor(text=prompt, images=[image1, image2])
-inputs = {k: torch.tensor(v).to(DEVICE) for k, v in inputs.items()}
+inputs = processor(text=prompt, images=[image1, image2], return_tensors="pt")
+inputs = {k: v.to(DEVICE) for k, v in inputs.items()}
 
 
 # Generate