Galuh
commited on
Commit
•
57929b3
1
Parent(s):
e7539a3
Update README.md
Browse files
README.md
CHANGED
|
@@ -52,7 +52,7 @@ def generate_step(batch):
|
|
| 52 |
return caption
|
| 53 |
|
| 54 |
image_file_path = image_file_path
|
| 55 |
-
image = read_image(
|
| 56 |
image = transforms(image)
|
| 57 |
pixel_values = torch.stack([image]).permute(0, 2, 3, 1).numpy()
|
| 58 |
|
|
|
|
| 52 |
return caption
|
| 53 |
|
| 54 |
image_file_path = image_file_path
|
| 55 |
+
image = read_image(image_file_path, mode=ImageReadMode.RGB)
|
| 56 |
image = transforms(image)
|
| 57 |
pixel_values = torch.stack([image]).permute(0, 2, 3, 1).numpy()
|
| 58 |
|