a

app.py

@@ -41,7 +41,7 @@ def get_caption_for_image_id(image_path):
     Retrieve the caption for a given image_id from the JSON data.
     """
     # Extract the numeric part of the image ID
-    match = re.search(r'(\d+)', image_path)
+    match = re.search(r'_(\d+)\.', image_path)
     if match:
         image_id = match.group(1).lstrip('0')  # Remove leading zeros
         print("Searching for image_id:", image_id)  # Debugging line