Update app.py

app.py

@@ -18,9 +18,13 @@ 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([index[ix][0]])["image"][0]
-    neighbors =  ds.select(index[ix][1:])
+    neighbors =  ds.select(index[ix][1:9])
     neighbor_images = neighbors["image"]
     neighbor_captions = [caption.split("/")[-1] for caption in neighbors["image_path"]]
+    Photo_portrait_of_a_Pacific_Islander_person_at_work_6.jpg
+    neighbor_captions = [' '.join(caption.split("_")[4:-3]) for caption in neighbor_captions]
+    neighbor_models = neighbors["model"]
+    neighbor_captions = [f"{a} {b}" for a,b in zip(neighbor_captions,neighbor_models)]
     return image, list(zip(neighbor_images, neighbor_captions))
 
 with gr.Blocks() as demo:
@@ -34,6 +38,6 @@ with gr.Blocks() as demo:
     button = gr.Button(value="Get nearest neighbors")
     with gr.Row():
         image = gr.Image()
-        gallery = gr.Gallery().style(grid=8)
+        gallery = gr.Gallery().style(grid=4)
     button.click(get_nearest_64, inputs=[gender, ethnicity, model, no], outputs=[image, gallery])
 demo.launch()