send height

streamlit_app.py

@@ -57,7 +57,7 @@ def display_message(role, content, image_path=None):
             # Display the image if it exists
             try:
                 img = Image.open(image_path)
-                st.image(img, caption="Generated Image", use_column_width=True)
+                st.image(img, caption="Generated Image", use_container_width=True)
             except Exception as e:
                 st.error(f"Error loading image: {e}")