Update my_model/results/demo.py

my_model/results/demo.py

@@ -209,7 +209,7 @@ class ResultDemonstrator:
         }
 
         for img_filename in target_imgs:
-            image_data = self.main_data[self.main_data['image_id'] == img_filename]
+            image_data = self.main_data[self.main_data['image_filename'] == img_filename]
             im = Image.open(f"{self.demo_images_path}/{img_filename}")
             col1, col2 = st.columns([1, 2])  # to display images side by side with their data.
             # Create a container for each image