Update app.py

app.py

@@ -555,5 +555,7 @@ else:
         cols = st.columns(4)
         for i, img_path in enumerate(st.session_state.contents):
             with cols[i % 4]:
-                st.image(img_path, caption=os.path.basename(img_path), use_column_width=True)
+                # use_column_width=True를 use_container_width=True로 변경
+                st.image(img_path, caption=os.path.basename(img_path), use_container_width=True)
+