dataset = load_dataset("sayakpaul/instructpix2pix-demo", split="train")
dataset.features
```  
```bash
{'input': Value(dtype='string', id=None),
'edit': Value(dtype='string', id=None),
'output': Value(dtype='string', id=None),
'image': Image(decode=True, id=None)}
```  
Here we have:  
- `input` is a caption corresponding to the `image`.
- `edit` denotes the edit instruction.
- `output` denotes the modified caption reflecting the `edit` instruction.  
Let's take a look at a sample.  
```python
idx = 0