Update my_model/tabs/dataset_analysis.py

my_model/tabs/dataset_analysis.py

@@ -246,17 +246,19 @@ class OKVQADatasetAnalyzer:
 
 
 def run_dataset_analyzer():
-    datasets_comparison_table = pd.read_excel("dataset_analyses.xlsx", sheet_name="VQA Datasets Comparison")
-    okvqa_dataset_characteristics = pd.read_excel("dataset_analyses.xlsx", sheet_name="OK-VQA Dataset Characteristics")
-    val_data = process_okvqa_dataset('OpenEnded_mscoco_val2014_questions.json', 'mscoco_val2014_annotations.json',
+    
+    datasets_comparison_table = pd.read_excel(CONFIG.DATASET_ANALYSES_PATH, sheet_name="VQA Datasets Comparison")
+    okvqa_dataset_characteristics = pd.read_excel(CONFIG.DATASET_ANALYSES_PATH, sheet_name="OK-VQA Dataset Characteristics")
+    
+    val_data = process_okvqa_dataset(config.DATASET_VAL_QUESTIONS_PATH, config.DATASET_VAL_ANNOTATIONS_PATH,
                                      save_to_csv=False)
-    train_data = process_okvqa_dataset('OpenEnded_mscoco_train2014_questions.json', 'mscoco_train2014_annotations.json',
+    train_data = process_okvqa_dataset(config.DATASET_TRAIN_QUESTIONS_PATH, config.DATASET_TRAIN_ANNOTATIONS_PATH ,
                                        save_to_csv=False)
 
 
 
-    dataset_analyzer = OKVQADatasetAnalyzer('OpenEnded_mscoco_train2014_questions.json',
-                                            'OpenEnded_mscoco_val2014_questions.json', 'train_test')
+    dataset_analyzer = OKVQADatasetAnalyzer(config.DATASET_TRAIN_QUESTIONS_PATH, 
+                                            config.DATASET_VAL_QUESTIONS_PATH, 'train_test')
 
     with st.container():
         st.markdown("## Overview of KB-VQA Datasets")