Update app.py

app.py

@@ -14,7 +14,7 @@ nos = [1,2,3,4,5,6,7,8,9,10]
 index = np.load("indexes/knn_10752_65.npy")
 ds = load_dataset("tti-bias/identities", split="train")
 
-def get_nearest_64(gender, ethnicity, model, no):
+def get_nearest_64(gender="man", ethnicity="Hispanic", model="SD_14", no=1):
     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]