Update app.py

app.py

@@ -15,6 +15,7 @@ index = np.load("indexes/knn_10752_65.npy")
 ds = load_dataset("SDBiaseval/identities", split="train")
 
 def get_index(gender, ethnicity, model, no):
+    df = ds.remove_columns(["image","image_path"]).to_pandas()
     return df.loc[(df['ethnicity'] == ethnicity)  & (df['gender'] == gender) & (df['no'] == no) & (df['model'] == model)].index[0]
 
 def get_nearest_64(gender, ethnicity, model, no):