Upload 2 files

app.py

@@ -187,7 +187,7 @@ def handle_userinput(user_question):
     
     with st.chat_message("assistant"):
         if has_math:
-            st.image(content, use_container_width=True)
+            st.image(content, use_column_width=True)
         else:
             st.markdown(content, unsafe_allow_html=True)
 
@@ -309,7 +309,7 @@ def main():
                 has_math, content = render_math_as_image(message.content)
                 with st.chat_message("assistant"):
                     if has_math:
-                        st.image(content, use_container_width=True)
+                        st.image(content, use_column_width=True)
                     else:
                         st.markdown(content, unsafe_allow_html=True)