Update my_model/results/demo.py

my_model/results/demo.py

@@ -87,7 +87,7 @@ class ResultDemonstrator:
 
     def display_main_results(self) -> None:
         """Displays the main model results from the Scores sheet, these are displayed from the file directly."""
-        main_scores = pd.read_excel('evaluation_results.xlsx', sheet_name="Scores", index_col=0)
+        main_scores = pd.read_excel(config.EVALUATION_DATA_PATH, sheet_name="Scores", index_col=0)
         st.markdown("### Main Model Results (Inclusive of Ablation Experiments)")
         main_scores.reset_index()
         self.display_table(main_scores)
@@ -208,7 +208,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"demo/{img_filename}")
+            im = Image.open(f"config.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():
@@ -223,7 +223,7 @@ class ResultDemonstrator:
                         st.text(image_data.iloc[0]['objects_yolov5'])
                 with col2:
                     if not image_data.empty:
-                        st.write(f"**Question: {image_data.iloc[0]['question']}**")
+                        st.write(f"**Question:** {image_data.iloc[0]['question']}")
                         st.write(f"**Ground Truth Answers:** {image_data.iloc[0]['raw_answers']}")
 
                         # Initialize an empty DataFrame for summary data