Update my_model/results/demo.py

my_model/results/demo.py

@@ -24,7 +24,7 @@ class ResultDemonstrator:
         """
         # Load data
         self.main_data = pd.read_excel(config.EVALUATION_DATA_PATH, sheet_name="Main Data")
-        self.sample_img_pool = list(os.listdir("Demo_Images"))
+        self.sample_img_pool = list(os.listdir(config.DEMO_IMAGES_PATH))
         self.model_names = config.MODEL_NAMES
         self.model_configs = config.MODEL_CONFIGURATIONS