Update my_model/utilities/ui_manager.py

my_model/utilities/ui_manager.py

@@ -1,5 +1,6 @@
-import pandas as pd
 import streamlit as st
+import streamlit.components.v1 as components
+import pandas as pd
 from my_model.tabs.run_inference import InferenceRunner
 from my_model.tabs.finetuning_evaluation import KBVQAEvaluator
 from my_model.state_manager import StateManager
@@ -68,8 +69,10 @@ class UIManager():
                         \n\n\nDeveloped by: [Mohammed H AlHaj](https://www.linkedin.com/in/m7mdal7aj)""")
         with col2:
             # Display the HTML content in Streamlit
-            st.markdown(model_arch_html, unsafe_allow_html=True)
+            #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)
             
         with col3:
             st.image("Files/mm.jpeg")