Update app.py

app.py

@@ -15,7 +15,7 @@ def main():
         if user_text.strip():
             # Generate image with text
             img = create_text_image(user_text)
-            st.image(img, caption="Generated Image", use_column_width=True)
+            st.image(img, caption="Generated Image", use_container_width=True)
         else:
             st.warning("Please enter some text before generating an image.")
 
@@ -46,7 +46,6 @@ def create_text_image(text):
     img_byte_arr.seek(0)
     return img_byte_arr
 
-
 # Run the app
 if __name__ == "__main__":
     main()