Update app.py

app.py

@@ -3,6 +3,7 @@ import pandas as pd
 from datasets import load_dataset
 import numpy as np
 
+
 gender_labels = ['man', 'non-binary', 'woman', 'no_gender_specified', ]
 
 ethnicity_labels = ['African-American', 'American_Indian', 'Black', 'Caucasian', 'East_Asian',
@@ -21,7 +22,7 @@ def get_nearest_64(gender, ethnicity, model, no):
     neighbors =  ds.select(index[ix][1:])
     neighbor_images = neighbors["image"]
     neighbor_captions = [caption.split("/")[-1] for caption in neighbors["image_path"]]
-    return image, list(zip(neighbor_images, neighbor_captions))
+    return image.resize(32,32), list(zip([i.resize(32,32) for i in neighbor_images], neighbor_captions))
 
 with gr.Blocks() as demo:
     with gr.Row():