Update app.py

app.py

@@ -36,7 +36,7 @@ def get_nearest(adjective, profession, model, no):
     neighbors =  ds.select(range(max(ix-10, 0), min(ix+10, len(ds)-1)))
     neighbor_images = neighbors["image"]
     neighbor_captions = [caption.split("/")[-1] for caption in neighbors["image_path"]]
-    neighbor_captions = [' '.join(caption.split("_")[4:-3]) for caption in neighbor_captions]
+    # 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)]
     return image, list(zip(neighbor_images, neighbor_captions))