import and install

app.py

@@ -1,4 +1,12 @@
 import os
+
+try:
+    import detectron2
+    import densepose
+except ImportError:
+    os.system('pip install -e ./preprocess/detectron2')
+    os.system('pip install -e ./preprocess/detectron2/projects/DensePose')
+
 import sys
 import time
 from pathlib import Path

requirements.txt

@@ -1,20 +1,18 @@
-torch==2.0.1
-torchvision==0.15.2
-torchaudio==2.0.2
-numpy==1.24.4
-scipy==1.10.1
-scikit-image==0.21.0
-opencv-python==4.7.0.72
-pillow==9.4.0
-diffusers==0.24.0
-transformers==4.36.2
-accelerate==0.26.1
-matplotlib==3.7.4
-tqdm==4.64.1
-gradio==4.16.0
-config==0.5.1
-einops==0.7.0
-onnxruntime==1.16.2
+torch
+torchvision
+numpy
+scipy
+scikit-image
+opencv-python
+pillow
+diffusers
+transformers
+accelerate
+matplotlib
+tqdm
+gradio
+einops
+onnxruntime
 basicsr
-./preprocess/detectron2
-./preprocess/detectron2/projects/DensePose
+# ./preprocess/detectron2
+# ./preprocess/detectron2/projects/DensePose