Update app.py

app.py

@@ -14,7 +14,7 @@ nos = [1,2,3,4,5,6,7,8,9,10]
 
 ds = load_from_disk("color-sorted")
 
-def get_nearest_64(gender, ethnicity, model, no):
+def get_nearest(gender, ethnicity, model, no):
     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([ix])["image"][0]
@@ -28,7 +28,7 @@ def get_nearest_64(gender, ethnicity, model, no):
 
 with gr.Blocks() as demo:
     gr.Markdown("# Colorfulness Nearest Neighbors Explorer")
-    gr.Markdown("### TF-IDF index of the _identities_ dataset of images generated by 3 models using colorfulness")
+    gr.Markdown("### Colorfulness 1-D index of the _identities_ dataset of images generated by 3 models")
     gr.Markdown("#### Choose one of the generated identity images to see its nearest neighbors according to colorfulness")
     with gr.Row():
         with gr.Column():
@@ -41,5 +41,5 @@ with gr.Blocks() as demo:
     with gr.Row():
         image = gr.Image()
         gallery = gr.Gallery().style(grid=4)
-    button.click(get_nearest_64, inputs=[gender, ethnicity, model, no], outputs=[image, gallery])
+    button.click(get_nearest, inputs=[gender, ethnicity, model, no], outputs=[image, gallery])
 demo.launch()