Update README.md

README.md

@@ -50,7 +50,7 @@ processor = AutoProcessor.from_pretrained(
 torch.cuda.empty_cache()
 
 image = Image.open("test.png")
-inputs = processor.process(images=img, text="Caption this image.")
+inputs = processor.process(images=image, text="Caption this image.")
 inputs = {k: v.to("cuda:0").unsqueeze(0) for k,v in inputs.items()}
 prompt_tokens = inputs["input_ids"].size(1)
 print("Prompt tokens:", prompt_tokens)