Update README.md

README.md

@@ -56,7 +56,7 @@ with torch.no_grad():
     pt_txt = tokenizer([text], return_tensors="pt").input_ids.cuda()
     pt_img = image_processor([image], return_tensors="pt").pixel_values.to(torch.float16).cuda()
     gen = model.generate(pt_txt, patch_images=pt_img, top_p=0.5, do_sample=True, no_repeat_ngram_size=3, max_length=256)
-print(tokenizer.batch_decode(gen, skip_special_tokens=True).replace("not yet.", ""))
+print(tokenizer.batch_decode(gen, skip_special_tokens=True)[0].replace("not yet.", ""))
 # e.g. [' is he the one who just threw the ball?']  # Due to the generator, different results may be output
 ```