Update app.py

app.py

@@ -20,7 +20,7 @@ def get_nearest_64(gender, ethnicity, model, no):
     image = ds.select([index[ix][0]])["image"][0]
     neighbors =  ds.select(index[ix][1:])
     neighbor_images = neighbors["image"]
-    neighbor_captions = neighbors["image_path"]
+    neighbor_captions = [caption.split("/")[-1] for caption in neighbors["image_path"]]
     return image, list(zip(neighbor_images, neighbor_captions))
 
 with gr.Blocks() as demo: