Update app.py

app.py

@@ -108,7 +108,7 @@ if page == "Dataset":
         with cols[idx]:
             st.write(f"### {label}")
             for img_path in images:
-                st.image(img_path, use_column_width=True)
+                st.image(img_path, use_container_width=True)
 
 # Data Visualization Page
 elif page == "Data Visualization":
@@ -148,7 +148,7 @@ elif page == "Model Metrics":
     
     # Confusion Matrix
     st.subheader("Confusion Matrix")
-    st.image("con_mat.png", use_column_width=True)
+    st.image("con_mat.png", use_container_width=True)
     
     # Training Curves
     col1, col2 = st.columns(2)
@@ -176,7 +176,7 @@ elif page == "Classification":
     
     if uploaded_file:
         image = Image.open(uploaded_file).convert("RGB")
-        st.image(image, use_column_width=True)
+        st.image(image, use_container_width=True)
         
         # Transform and predict
         image_tensor = transform(image).unsqueeze(0)