Update deployment with latest changes

app/main.py

@@ -21,8 +21,9 @@ app.add_middleware(
 )
 
 class ModelName(str, Enum):
+    birefnet_general = "birefnet-general"
+    birefnet_general_lite = "birefnet-general-lite"
     isnet_anime = "isnet-anime"
-    isnet_general_use = "isnet-general-use"
     u2net = "u2net"
 
 # Cache sessions to avoid reloading models on every request
@@ -35,8 +36,8 @@ def get_session(model_name: str):
     return sessions[model_name]
 
 # Pre-load the default model suitable for Mascots/Cartoons
-# 'isnet-anime' is generally best for high-contrast anime/cartoon characters
-DEFAULT_MODEL = ModelName.isnet_anime
+# 'birefnet-general' offers superior edge detection and quality for mascots
+DEFAULT_MODEL = ModelName.birefnet_general
 
 @app.on_event("startup")
 async def startup_event():

requirements.txt

@@ -7,4 +7,6 @@ numpy
 pillow
 aiofiles
 pyngrok
+rembg
 onnxruntime
+tqdm