edit streamlit version

requirements.txt

@@ -7,6 +7,6 @@ peft>=0.11.0
 torch>=2.0.0
 torchvision>=0.15.0
 Pillow>=10.0.0
-streamlit==1.28.2
+streamlit>=1.50
 numpy>=1.24.0
 scipy>=1.11.0

src/app.py

@@ -39,16 +39,6 @@ from model.generator import PepeGenerator
 from model.config import ModelConfig
 from utils.image_processor import ImageProcessor
 
-def supports_use_container_width():
-    """Check if the current Streamlit version supports use_container_width parameter."""
-    try:
-        import pkg_resources
-        streamlit_version = pkg_resources.get_distribution("streamlit").version
-        # use_container_width was introduced in Streamlit 1.16.0
-        return tuple(map(int, streamlit_version.split('.')[:2])) >= (1, 16)
-    except:
-        return False
-
 # Page config
 st.set_page_config(
     page_title="🐸 Pepe Meme Generator",
@@ -296,14 +286,9 @@ def main():
         placeholder = st.empty()
         
         if st.session_state.generated_images:
-            # Use use_container_width only if supported by Streamlit version
-            image_kwargs = {}
-            if supports_use_container_width():
-                image_kwargs['use_container_width'] = True
-
             placeholder.image(
                 st.session_state.generated_images[-1],
-                **image_kwargs
+                use_column_width=True
             )
         else:
             placeholder.info("Your meme will appear here...")
@@ -362,12 +347,7 @@ def main():
             
             # Show result
             if num_vars == 1:
-                # Use use_container_width only if supported by Streamlit version
-                image_kwargs = {}
-                if supports_use_container_width():
-                    image_kwargs['use_container_width'] = True
-
-                placeholder.image(image, **image_kwargs)
+                placeholder.image(image, use_column_width=True)
                 
                 # Download
                 buf = io.BytesIO()
@@ -383,12 +363,7 @@ def main():
                 cols = st.columns(min(num_vars, 2))
                 for idx, img in enumerate(st.session_state.generated_images[-num_vars:]):
                     with cols[idx % 2]:
-                        # Use use_container_width only if supported by Streamlit version
-                        image_kwargs = {}
-                        if supports_use_container_width():
-                            image_kwargs['use_container_width'] = True
-
-                        st.image(img, **image_kwargs)
+                        st.image(img, use_column_width=True)
         
         except Exception as e:
             st.error(f"Error: {str(e)}")
@@ -403,12 +378,7 @@ def main():
             cols = st.columns(4)
             for idx, img in enumerate(reversed(st.session_state.generated_images[-8:])):
                 with cols[idx % 4]:
-                    # Use use_container_width only if supported by Streamlit version
-                    image_kwargs = {}
-                    if supports_use_container_width():
-                        image_kwargs['use_container_width'] = True
-
-                    st.image(img, **image_kwargs)
+                    st.image(img, use_column_width=True)
     
     # Footer
     st.divider()