Update my_model/utilities/ui_manager.py

my_model/utilities/ui_manager.py

@@ -72,7 +72,8 @@ class UIManager():
             #st.markdown(model_arch_html, unsafe_allow_html=True)
             #st.image("Files/Model Arch.png")
             # Display the HTML content in Streamlit
-            components.html(model_arch_html, height=800)
+            st.header("Model Architecture")
+            components.html(model_arch_html, height=1500)
             
         with col3:
             st.image("Files/mm.jpeg")