Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from rembg import remove
+from rembg import remove, new_session
 from PIL import Image
 import io
 import fitz  # PyMuPDF for handling PDFs
@@ -25,9 +25,12 @@ if uploaded_file is not None:
 
     st.image(image, caption="Original Image", use_column_width=True)
 
-    # Remove background using HQ model
+    # Create a high-quality model session
+    session = new_session("u2net_hq")  # Load the HQ model
+
+    # Remove background using the HQ model
     with st.spinner("Removing background with high quality..."):
-        output_image = remove(image, model_name="u2net_hq")  # Use the HQ model
+        output_image = remove(image, session=session)
 
     st.image(output_image, caption="Background Removed", use_column_width=True)
 
@@ -37,4 +40,4 @@ if uploaded_file is not None:
     st.download_button("Download Image", img_byte_arr.getvalue(), file_name="no_bg_image.png", mime="image/png")
 
 st.markdown("---")
-st.markdown("💡 Developed with ❤️ using Streamlit and rembg (U2Net HQ)")
+st.markdown("💡 Developed with ❤️ using Streamlit and rembg (U2Net HQ)")