Update my_model/tabs/dataset_analysis.py

my_model/tabs/dataset_analysis.py

@@ -300,18 +300,17 @@ def run_dataset_analyzer():
         st.write("\n" * 10)
         st.markdown("## OK-VQA Dataset")
         st.write("This model was fine-tuned and evaluated using OK-VQA dataset.\n")
-        col1, col2, col3 = st.columns([2, 5, 5])
 
-        with col1:
+        with st.expander("OK-VQA Dataset Characteristics"):
             st.markdown("#### OK-VQA Dataset Characteristics")
             st.write(okvqa_dataset_characteristics)
-        with col2:
+        with st.expander("Questions Distribution over Knowledge Category"):
             df = pd.read_excel(config.DATASET_ANALYSES_PATH, sheet_name="Question Category Dist")
             st.markdown("#### Questions Distribution over Knowledge Category")
             dataset_analyzer.plot_bar_chart(df, "Knowledge Category", "Percentage", "Questions Distribution over "
                                                                                       "Knowledge Category")
 
-        with col3:
+        with st.expander("Distribution of Question Keywords"):
 
             #with st.expander("Distribution of Question Keywords"):
             dataset_analyzer.categorize_questions()