Update requirements.txt

requirements.txt

@@ -1,27 +1,23 @@
-# BEN2 Background Removal - Base Requirements
-# Install these first, then use the app to install BEN2 or rembg
+# Official InsPyReNet Requirements
+# This will install the official implementation from the ACCV 2022 paper authors
 
-# Core dependencies - INSTALL THESE FIRST
+# Official InsPyReNet package (contains the full model)
+transparent-background
+
+# Core dependencies
 gradio>=4.0.0
 torch>=1.12.0
 torchvision>=0.13.0
 Pillow>=9.0.0
 numpy>=1.21.0
 requests>=2.25.0
-opencv-python>=4.5.0
 
-# Basic ML dependencies
-transformers>=4.21.0
-
-# The app will install ONE of these via the UI buttons:
-# 
-# Option 1 (Best Quality): BEN2
-# git+https://github.com/PramaLLC/BEN2.git
-# 
-# Option 2 (Good Fallback): rembg
-# rembg
-#
-# Option 3 (Basic): Built-in OpenCV method (no extra install needed)
+# Additional dependencies for the official model
+opencv-python>=4.5.0
+scipy>=1.7.0
+matplotlib>=3.5.0
 
-# Optional: CUDA support for faster processing
-# torch==2.1.0+cu118 torchvision==0.16.0+cu118 --index-url https://download.pytorch.org/whl/cu118
+# For CUDA support (optional, will use CPU if not available)
+# Uncomment the appropriate line based on your CUDA version:
+# torch==2.1.0+cu118 torchvision==0.16.0+cu118 --index-url https://download.pytorch.org/whl/cu118
+# torch==2.1.0+cu121 torchvision==0.16.0+cu121 --index-url https://download.pytorch.org/whl/cu121