Update app.py

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 import os
 import zipfile
 from PIL import Image, ImageDraw, ImageFont
+from rembg import remove
+
 
 def draw_ruler(image, dpi=300, unit="in", color="black"):
     """
@@ -219,7 +221,7 @@ def main():
                 if index < num_images:
                     cols[col].subheader(f"Image {index+1}")
                     cols[col].image(uploaded_files[index], caption="Original Image", width=col_width)
-                    cols[col].image(cropped_images[index], caption="Cropped Image", width=col_width)
+                    cols[col].image(remove(cropped_images[index]), caption="Cropped Image", width=col_width)
 
         # Save all cropped images
         if st.button("Save All"):