import gradio as gr
from datasets import load_dataset
import numpy as np

index = np.load("index/knn_10752_65.npy")
ds = load_dataset("SDBiaseval/identities", split="train")

def get_nearest_64(ix):
    return ds.select(index[ix][1:])