Sorting height problems

main.py

@@ -25,7 +25,7 @@ def main():
         for img_file in img_files or []:
             img_bytes = img_file.read()
             caption = img_caption.get_caption(img_bytes)
-            st.image(image=img_bytes, caption=caption, width=250, height=250)
+            st.image(image=img_bytes, caption=caption, width=250)
             generated_topics = topic_generator.generate_topics(caption)
             for idx, topic in enumerate(generated_topics, 1):
                 st.write(f"Topic {idx}: {topic}")
@@ -35,7 +35,7 @@ def main():
         if url_input:
             img_load = load_image(check_url(url_input))
             caption = img_caption.get_caption(img_load)
-            st.image(image=img_load, caption=caption, width=250, height=250)
+            st.image(image=img_load, caption=caption, width=250)
             generated_topics = topic_generator.generate_topics(caption)
             for idx, topic in enumerate(generated_topics, 1):
                 st.write(f"Topic {idx}: {topic}")