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[0])["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"]]