if __name__ == "__main__":
    import datasets
    from renumics import spotlight

    dataset = datasets.load_dataset("renumics/cifar100-enriched", split="train")
    
    df = dataset.to_pandas()
    df_show = df.drop(columns=['embedding', 'probabilities'])
    spotlight.show(df_show, port=7860, host="0.0.0.0", dtype={"image": spotlight.Image, "embedding_reduced": spotlight.Embedding})