Update app.py

app.py

@@ -1,8 +1,11 @@
 import streamlit as st
-from rembg import remove
+from rembg import remove, new_session
 from PIL import Image, ImageFilter
 import io
 
+# Load high-quality model
+session = new_session("u2net")
+
 # Streamlit App Configuration
 st.set_page_config(page_title="Background Remover & Changer", layout="wide")
 st.title("Background Remover & Changer")
@@ -21,7 +24,7 @@ if uploaded_files:
         image = image.convert("RGBA")
         
         # Remove background using high-quality model
-        removed_bg = remove(image, model_name="u2net")
+        removed_bg = remove(image, session=session)
         
         # Apply smoothing to improve edges
         removed_bg = removed_bg.filter(ImageFilter.SMOOTH)