Update app.py

app.py

@@ -18,7 +18,7 @@ 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]
-    neighbors =  ds.select(max(ix-10, 0), min(ix+10, len(ds)-1))
+    neighbors =  ds.select(range(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"]]
     neighbor_captions = [' '.join(caption.split("_")[4:-3]) for caption in neighbor_captions]