Update app.py

app.py

@@ -42,7 +42,7 @@ def get_top_prediction(predictions):
 if uploaded_file is not None:
     # Process uploaded image
     image = Image.open(uploaded_file)
-    st.image(image, caption="Uploaded Image", use_column_width=True)
+    st.image(image, caption="Uploaded Image", use_container_width=True)
     
     with st.spinner("Analyzing age..."):
         predictions = pipe(image)
@@ -61,7 +61,7 @@ elif 'example_loaded' in st.session_state and st.session_state.example_loaded:
     with st.spinner("Loading example image..."):
         # Download and load the image properly
         image = load_image_from_url(st.session_state.example_image)
-        st.image(image, caption="Example Image", use_column_width=True)
+        st.image(image, caption="Example Image", use_container_width=True)
     
     with st.spinner("Analyzing age..."):
         # Pass the actual PIL Image object to the pipeline