from datasets import load_dataset
from IPython.display import display
from PIL import Image
dataset = load_dataset("gaodrew/gemstone_eval-4by4-256by256px-4shapes-8colors-1k", split='train')
#shapes = ["circle", "triangle", "square", "star"]
#colors = ["red", "orange", "yellow", "green", "blue", "purple", "pink", "black"]
print(dataset)
Display the first few images
for i in range(10):
image_path = dataset[i]['image']
print(f"Image {i+1}:")
display(Image.open(image_path))