Update app.py

app.py

@@ -50,8 +50,8 @@ st.markdown(
     }
     .group-banner {
         text-align: center;
-        font-size: 0.9em;
-        color: #777777;
+        font-size: 1.5em;
+        color: #ffffff;
         font-weight: bold;
         margin-top: 2em;
     }
@@ -83,7 +83,7 @@ if uploaded_file is not None:
         with col1:
             st.markdown("### ⬆ Original")
             original_img = Image.open(uploaded_file)
-            st.image(original_img, use_column_width=True)
+            st.image(original_img, use_container_width=True)
         
         with col2:
             st.markdown("### 🎨 Colorized")
@@ -101,7 +101,7 @@ if uploaded_file is not None:
                     colorized_image = inverse_transform_cs(result.squeeze(0), original_size)
                     
                     # Display result
-                    st.image(colorized_image, use_column_width=True)
+                    st.image(colorized_image, use_container_width=True)
                     st.success("Colorization complete!")
 
 # Footer