Update app.py

app.py

@@ -17,9 +17,11 @@ ds = load_dataset("SDBiaseval/identities", split="train")
 def get_nearest_64(gender, ethnicity, model, no):
     df = ds.remove_columns(["image","image_path"]).to_pandas()
     ix = df.loc[(df['ethnicity'] == ethnicity)  & (df['gender'] == gender) & (df['no'] == no) & (df['model'] == model)].index[0]
-    image = ds.select([index[ix][0]])["image"]
-    neighbor_list = ds.select(index[ix][1:])["image"]
-    return image, neighbor_list 
+    image = ds.select([index[ix][0]])["image"][0]
+    neighbors =  ds.select(index[ix][1:])
+    neighbor_images = neighbors["image"]
+    neighbor_captions = neighbors["image_path"]
+    return image, list(zip(neighbor_images, neighbor_captions))
 
 with gr.Blocks() as demo:
     gender = gr.Radio(gender_labels, label="Gender label")