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)
|