Create app.py

app.py

@@ -0,0 +1,30 @@
+from io import BytesIO
+
+import streamlit as st
+from PIL import Image
+from rembg import remove
+
+st.title("Hello Upload!")
+
+# Upload the file
+image_upload = st.file_uploader("Upload an image", type=["png", "jpg", "jpeg"])
+
+# Convert the image to BytesIO so we can download it!
+def convert_image(img):
+    buf = BytesIO()
+    img.save(buf, format="PNG")
+    byte_im = buf.getvalue()
+    return byte_im
+
+# If we've uploaded an image, open it and remove the background!
+if image_upload:
+    # SHOW the uploaded image!
+    st.image(image_upload)
+    image = Image.open(image_upload)
+    fixed = remove(image)
+    downloadable_image = convert_image(fixed)
+    # SHOW the improved image!
+    st.image(downloadable_image)
+    st.download_button(
+        "Download fixed image", downloadable_image, "fixed.png", "image/png"
+    )