Update app.py

app.py

@@ -288,7 +288,7 @@ def main():
         if uploaded_file is not None:
             try:
                 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..."):
                     img_tensor = preprocess_image(image)
@@ -305,7 +305,7 @@ def main():
         if camera_photo is not None:
             try:
                 image = Image.open(camera_photo)
-                st.image(image, caption="Camera Photo", use_column_width=True)
+                st.image(image, caption="Camera Photo", use_container_width=True)
                 
                 with st.spinner("🧠 Analyzing..."):
                     img_tensor = preprocess_image(image)