import os os.environ["NUMBA_DISABLE_CACHE"] = "1" import streamlit as st from rembg import remove from PIL import Image import io # App Title st.title("🧼 Background Remover App") st.markdown("Upload an image, and we'll remove the background using [rembg](https://github.com/danielgatis/rembg).") # File uploader uploaded_file = st.file_uploader("Upload an image", type=["png", "jpg", "jpeg"]) # Remove background button if uploaded_file is not None: image = Image.open(uploaded_file) st.image(image, caption="Original Image", use_column_width=True) if st.button("Remove Background"): with st.spinner("Removing background..."): # Remove background result = remove(image) # Display result st.image(result, caption="Image Without Background", use_column_width=True) # Convert result to bytes for download buf = io.BytesIO() result.save(buf, format="PNG") byte_im = buf.getvalue() st.download_button( label="📥 Download Transparent Image", data=byte_im, file_name="output.png", mime="image/png" )