Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import pandas as pd
 from datasets import load_dataset
 import numpy as np
 
@@ -14,8 +15,11 @@ index = np.load("indexes/knn_10752_65.npy")
 ds = load_dataset("SDBiaseval/identities", split="train")
 
 def get_index(gender, ethnicity, model, no):
-    pass
+    return df.loc[(df['ethnicity'] == ethnicity)  & (df['gender'] == gender) & (df['no'] == no) & (df['model'] == model)].index[0]
 
-def get_nearest_64(ix):
-    return ds.select(index[ix][1:])
+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"]
 
+# demo = gr.Interface(fn=query_db, inputs="text", outputs=[gr.Image(), gr.Gallery().style(grid=[8])])
+# demo.launch(debug=True)