Update my_model/results/demo.py

my_model/results/demo.py

@@ -16,6 +16,7 @@ class ResultDemonstrator:
             sample_img_pool (list[str]): List of image file names available for demonstration.
             model_names (list[str]): List of model names as defined in the configuration.
             model_configs (list[str]): List of model configurations as defined in the configuration.
+            demo_images_path(str): Path to the demo images directory.
     """
 
     def __init__(self) -> None:
@@ -27,6 +28,7 @@ class ResultDemonstrator:
         self.sample_img_pool = list(os.listdir(config.DEMO_IMAGES_PATH))
         self.model_names = config.MODEL_NAMES
         self.model_configs = config.MODEL_CONFIGURATIONS
+        self.demo_images_path = config.DEMO_IMAGES_PATH
 
     @staticmethod
     def display_table(data: pd.DataFrame) -> None:
@@ -208,7 +210,7 @@ class ResultDemonstrator:
 
         for img_filename in target_imgs:
             image_data = self.main_data[self.main_data['image_filename'] == img_filename]
-            im = Image.open(f"{config.DEMO_IMAGES_PATH}/{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
             with st.container():