Galuh commited on
Commit
57929b3
1 Parent(s): e7539a3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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('000000039769.jpg', mode=ImageReadMode.RGB)
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