Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from datasets import load_from_disk
 import numpy as np
+import re
 
 
 adjectives = ['ambitious', 'assertive', 'committed', 'compassionate', 'confident', 'considerate', 'decisive', 'determined', 'emotional', 'gentle', 
@@ -40,7 +41,7 @@ def get_nearest(adjective, profession, model, no):
     neighbor_captions = [caption.split("/")[-1] for caption in neighbors["image_path"]]
     # neighbor_captions = [' '.join(caption.split("_")[4:-3]) for caption in neighbor_captions]
     neighbor_models = neighbors["model"]
-    neighbor_captions = [f"{a} {b}" for a,b in zip(neighbor_captions,neighbor_models)]
+    neighbor_captions = [f"{re.split("Photo_portrait_of_an?_", a)[-1]} {b}" for a,b in zip(neighbor_captions,neighbor_models)]
     return image, list(zip(neighbor_images, neighbor_captions))
 
 with gr.Blocks() as demo: