Spaces:
Runtime error
Runtime error
Anonymous Authors
commited on
Commit
·
29170e8
1
Parent(s):
1cea8e8
Update app.py
Browse files
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
|
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]
|