MagiBoss commited on
Commit
746d582
1 Parent(s): ca5dc67

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -47,7 +47,7 @@ image = Image.open("Your image...").convert("RGB")
47
 
48
  # Generate a caption
49
  inputs = processor(images=image, return_tensors="pt", padding=True).to(device, torch.bfloat16)
50
- outputs = model.generate(**inputs, max_length=30, pad_token_id=processor.tokenizer.pad_token_id)
51
  caption = processor.batch_decode(outputs, skip_special_tokens=True)
52
 
53
  print("Generated Caption:", caption)
 
47
 
48
  # Generate a caption
49
  inputs = processor(images=image, return_tensors="pt", padding=True).to(device, torch.bfloat16)
50
+ outputs = model.generate(**inputs, max_length=30)
51
  caption = processor.batch_decode(outputs, skip_special_tokens=True)
52
 
53
  print("Generated Caption:", caption)