refactored app.py

app.py

@@ -160,7 +160,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
@@ -210,7 +210,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