Update app.py

app.py

@@ -25,7 +25,7 @@ if uploaded_file is not None:
     image = Image.open(uploaded_file).convert("RGB")
 
     # Display uploaded image
-    st.image(image, caption="Uploaded Image", use_column_width=True)
+    st.image(image, caption="Uploaded Image", use_container_width=True)
 
     # Get prediction
     predicted_age, confidence = classify_age(image)