Update app.py

app.py

@@ -17,7 +17,7 @@ ds = load_from_disk("color-sorted")
 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(ix)["image"][0]
+    image = ds.select([ix])["image"][0]
     neighbors =  ds.select(max(ix-10, 0), min(ix+10, len(ds)-1))
     neighbor_images = neighbors["image"]
     neighbor_captions = [caption.split("/")[-1] for caption in neighbors["image_path"]]
@@ -34,9 +34,9 @@ with gr.Blocks() as demo:
         with gr.Column():
             model = gr.Radio(models, label="Model")
             gender = gr.Radio(gender_labels, label="Gender label")
+            no = gr.Radio(nos, label="Image number")
         with gr.Column():
             ethnicity = gr.Radio(ethnicity_labels, label="Ethnicity label")
-            no = gr.Radio(nos, label="Image number")
     button = gr.Button(value="Get nearest neighbors")
     with gr.Row():
         image = gr.Image()