Update my_model/tabs/model_arch.py

my_model/tabs/model_arch.py

@@ -6,19 +6,18 @@ def run_model_arch() -> None:
     """
     Displays the model architecture and accompanying abstract and design details for the Knowledge-Based Visual Question
     Answering (KB-VQA) model.
-
     This function reads an HTML file containing the model architecture and renders it in a Streamlit application.
     It also provides detailed descriptions of the research, abstract, and design of the KB-VQA model.
-
+    
     Returns:
         None
     """
 
     # Read the model architecture HTML file
-    with open("Files/Model Arch.html", 'r', encoding='utf-8') as f:
+    with open("Model Arch.html", 'r', encoding='utf-8') as f:
         model_arch_html = f.read()
 
-    col1, col2 = st.columns([1,1])
+    col1, col2 = st.columns(2)
     with col1:
         st.markdown("#### Model Architecture")
         components.html(model_arch_html, height=1400)
@@ -53,7 +52,8 @@ def run_model_arch() -> None:
         and successfully retrieves the necessary knowledge to answer visual questions.
         </div>
         """, unsafe_allow_html=True)
-        st.write("\n"*3)
+
+        st.markdown("<br>" * 4, unsafe_allow_html=True)
         st.markdown("#### Design")
         st.markdown("""
         <div style="text-align: justify;">
@@ -69,4 +69,5 @@ def run_model_arch() -> None:
         (PT-LLMs), which is responsible for performing reasoning and deriving the required knowledge to formulate an 
         informed response to the question.
         </div>
-        """, unsafe_allow_html=True)
+        """, unsafe_allow_html=True)
+