Anonymous Authors commited on
Commit
29170e8
·
1 Parent(s): 1cea8e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ nos = [1,2,3,4,5,6,7,8,9,10]
14
  indexes = [768, 1536, 10752]
15
  ds = load_dataset("tti-bias/identities", split="train")
16
 
17
- def get_nearest_64(gender=, ethnicity, model, no, index):
18
  df = ds.remove_columns(["image","image_path"]).to_pandas()
19
  index = np.load(f"indexes/knn_{index}_65.npy")
20
  ix = df.loc[(df['ethnicity'] == ethnicity) & (df['gender'] == gender) & (df['no'] == no) & (df['model'] == model)].index[0]
 
14
  indexes = [768, 1536, 10752]
15
  ds = load_dataset("tti-bias/identities", split="train")
16
 
17
+ def get_nearest_64(gender, ethnicity, model, no, index):
18
  df = ds.remove_columns(["image","image_path"]).to_pandas()
19
  index = np.load(f"indexes/knn_{index}_65.npy")
20
  ix = df.loc[(df['ethnicity'] == ethnicity) & (df['gender'] == gender) & (df['no'] == no) & (df['model'] == model)].index[0]