Update README.md
Browse files
README.md
CHANGED
|
@@ -24,3 +24,13 @@ img1_resized = img1.resize((int(0.3 * width), int(0.3 * height))
|
|
| 24 |
|
| 25 |
display(img1_resized)
|
| 26 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
|
| 25 |
display(img1_resized)
|
| 26 |
|
| 27 |
+
# testing image
|
| 28 |
+
inputs = processor(images=img1, return_tensors="pt")
|
| 29 |
+
|
| 30 |
+
pixel_values = inputs.pixel_values
|
| 31 |
+
|
| 32 |
+
generated_ids = model.generate(pixel_values=pixel_values, max_length=50)
|
| 33 |
+
|
| 34 |
+
generated_caption = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
|
| 35 |
+
|
| 36 |
+
print(generated_caption)
|