Update app.py

app.py

@@ -2,8 +2,20 @@ import gradio as gr
 from datasets import load_dataset
 import numpy as np
 
+gender_labels = ['man', 'no_gender_specified', 'non-binary', 'woman']
+
+ethnicity_labels = ['African-American', 'American_Indian', 'Black', 'Caucasian', 'East_Asian',
+                    'First_Nations', 'Hispanic', 'Indigenous_American', 'Latino', 'Latinx',
+                    'Multiracial', 'Native_American', 'Pacific_Islander', 'South_Asian',
+                    'Southeast_Asian', 'White', 'no_ethnicity_specified']
+models = ['DallE', 'SD_14', 'SD_2']
+
 index = np.load("indexes/knn_10752_65.npy")
 ds = load_dataset("SDBiaseval/identities", split="train")
 
+def get_index(gender, ethnicity, model, no):
+    pass
+
 def get_nearest_64(ix):
     return ds.select(index[ix][1:])
+