changes made to all the files

app.py

@@ -177,7 +177,7 @@ if standard_button:
             caption = f"Standard Stable Diffusion"
             if concept_name:
                 caption += f" with {concept_name} concept"
-            st.image(image, caption=caption, use_column_width=True)
+            st.image(image, caption=caption, use_container_width=True)
             st.write(f"Generation time: {end_time - start_time:.2f} seconds")
 
 # Generate color-guided image
@@ -227,7 +227,7 @@ if color_button:
             caption = f"Color-Guided Stable Diffusion"
             if concept_name:
                 caption += f" with {concept_name} concept"
-            st.image(image, caption=caption, use_column_width=True)
+            st.image(image, caption=caption, use_container_width=True)
             st.write(f"Generation time: {end_time - start_time:.2f} seconds")
 
 # Explanation section