Update app.py

app.py

@@ -20,7 +20,7 @@ def get_index(gender, ethnicity, model, no):
 
 def get_nearest_64(gender, ethnicity, model, no):
     ix = get_index(gender, ethnicity, model, no)
-    return ds.select(index[ix][0])["image"], ds.select(index[ix][1:])["image"]
+    return ds.select([index[ix][0]])["image"], ds.select(index[ix][1:])["image"]
 
 with gr.Blocks() as demo:
     gender = gr.Radio(gender_labels, label="Gender label")