Update app.py

app.py

@@ -61,13 +61,10 @@ def display_images(image_collection, query_text, max_distance=None, debug=False)
             try:
                 img = PILImage.open(uri)
                 with cols[i % 3]:  # Use modulo to cycle through columns
-                    st.image(img, use_column_width=True)
+                    st.image(img, use_container_width = True)
             except Exception as e:
                 st.error(f"Error loading image: {e}")
 
-
-            
-
 def display_videos_streamlit(video_collection, query_text, max_distance=None, max_results=5, debug=False):
     """
     Display videos in a Streamlit app based on a query.