refactor: Delegate Python package installation to `requirements.txt` and streamline checkpoint download.

app.py

@@ -12,28 +12,18 @@ import threading
 from glob import glob
 
 # --- Installation Helper ---
-def install_dependencies():
-    print("Checking and installing dependencies...")
-    try:
-        # Upgrade build tools first
-        subprocess.check_call([sys.executable, "-m", "pip", "install", "--upgrade", "pip", "wheel", "setuptools", "ninja"])
-        
-        # Install Detectron2
-        subprocess.check_call([sys.executable, "-m", "pip", "install", "--no-build-isolation", "detectron2@git+https://github.com/facebookresearch/detectron2.git"])
-        # Install ROMP
-        subprocess.check_call([sys.executable, "-m", "pip", "install", "--no-build-isolation", "git+https://github.com/ZaiqiangWu/ROMP.git#subdirectory=simple_romp"])
-        
-        # Download Checkpoints if not present
-        if not os.path.exists("./rtv_ckpts"):
-            print("Downloading checkpoints...")
+# Dependencies are handled by requirements.txt on Spaces
+
+def check_chkpts():
+    if not os.path.exists("./rtv_ckpts"):
+        print("Downloading checkpoints...")
+        try:
             subprocess.check_call(["git", "clone", "https://huggingface.co/wuzaiqiang/rtv_ckpts"])
-        
-        print("Dependencies installed and checkpoints ready.")
-    except Exception as e:
-        print(f"Error installing dependencies: {e}")
+            print("Checkpoints downloaded.")
+        except Exception as e:
+            print(f"Error downloading checkpoints: {e}")
 
-# Run installation on startup
-install_dependencies()
+check_chkpts()
 
 # --- App Logic ---
 

requirements.txt

@@ -3,9 +3,11 @@ opencv-python-headless
 onnxruntime
 wheel
 ninja
-setuptools
+setuptools<70
 torch==2.4.0
 torchvision==0.19.0
+git+https://github.com/facebookresearch/detectron2.git
+git+https://github.com/ZaiqiangWu/ROMP.git#subdirectory=simple_romp
 rembg
 openmim
 numpy~=1.24.2