Update README.md
Browse files
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
|
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)
|