Ali Mohsin commited on
Commit
18b2585
·
1 Parent(s): b867451

chore: Add upgrade for pip, wheel, setuptools, and ninja before dependency installation.

Browse files
Files changed (2) hide show
  1. app.py +3 -0
  2. requirements.txt +3 -0
app.py CHANGED
@@ -15,6 +15,9 @@ from glob import glob
15
  def install_dependencies():
16
  print("Checking and installing dependencies...")
17
  try:
 
 
 
18
  # Install Detectron2
19
  subprocess.check_call([sys.executable, "-m", "pip", "install", "--no-build-isolation", "detectron2@git+https://github.com/facebookresearch/detectron2.git"])
20
  # Install ROMP
 
15
  def install_dependencies():
16
  print("Checking and installing dependencies...")
17
  try:
18
+ # Upgrade build tools first
19
+ subprocess.check_call([sys.executable, "-m", "pip", "install", "--upgrade", "pip", "wheel", "setuptools", "ninja"])
20
+
21
  # Install Detectron2
22
  subprocess.check_call([sys.executable, "-m", "pip", "install", "--no-build-isolation", "detectron2@git+https://github.com/facebookresearch/detectron2.git"])
23
  # Install ROMP
requirements.txt CHANGED
@@ -1,6 +1,9 @@
1
  gradio
2
  opencv-python-headless
3
  onnxruntime
 
 
 
4
  torch==2.4.0
5
  torchvision==0.19.0
6
  rembg
 
1
  gradio
2
  opencv-python-headless
3
  onnxruntime
4
+ wheel
5
+ ninja
6
+ setuptools
7
  torch==2.4.0
8
  torchvision==0.19.0
9
  rembg