aptol commited on
Commit
e819ff1
·
verified ·
1 Parent(s): 893125e

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +14 -18
requirements.txt CHANGED
@@ -1,8 +1,8 @@
1
- # UI
2
  gradio==4.44.0
3
  gradio_client>=1.3.0
4
 
5
- # Core (※ ZeroGPU에서 torch는 베이스 이미지 사용)
6
  numpy==1.26.4
7
  Pillow
8
  scipy
@@ -12,36 +12,32 @@ huggingface_hub
12
  safetensors
13
  accelerate==0.31.0
14
 
15
- # Diffusers / ControlNet
16
  diffusers==0.29.2
17
  transformers==4.43.4
18
  opencv-python-headless==4.10.0.84
19
  imageio
20
  scikit-image
21
 
22
- # Background removal
23
  rembg==2.0.56
24
  onnxruntime==1.18.0
25
 
26
- # Weapon removal (inpaint)
27
- simple-lama-inpainting
 
28
 
29
- # Hands (21 keypoints)
30
- mediapipe==0.10.14
31
- protobuf<5
32
- lama-cleaner
33
- # 3D & optional 6-view render
34
  trimesh
35
- networkx
36
  pyrender
37
  PyOpenGL
38
  PyOpenGL-accelerate
39
  pyglet
40
- groundingdino
41
- segment-anything
42
  controlnet-aux
43
 
44
- # --- GroundingDINO / SAM: PyPI에 없음 → Git 소스에서 설치 ---
45
- git+https://github.com/IDEA-Research/GroundingDINO.git@main
46
- git+https://github.com/facebookresearch/segment-anything.git
47
- pycocotools>=2.0.7
 
1
+ # --- Core UI/runtime ---
2
  gradio==4.44.0
3
  gradio_client>=1.3.0
4
 
5
+ # --- Sci/IO ---
6
  numpy==1.26.4
7
  Pillow
8
  scipy
 
12
  safetensors
13
  accelerate==0.31.0
14
 
15
+ # --- Diffusers / Transformers ---
16
  diffusers==0.29.2
17
  transformers==4.43.4
18
  opencv-python-headless==4.10.0.84
19
  imageio
20
  scikit-image
21
 
22
+ # --- Background removal ---
23
  rembg==2.0.56
24
  onnxruntime==1.18.0
25
 
26
+ # --- Inpainting (둘 중 하나만 사용) ---
27
+ lama-cleaner==1.2.5 # ← 코드가 lama_cleaner.ModelManager를 쓰면 이거만 남기세요
28
+ # simple-lama-inpainting # ← 쓰지 않으면 주석 처리(둘 다 넣지 마세요)
29
 
30
+ # --- 3D 렌더/출력 ---
 
 
 
 
31
  trimesh
 
32
  pyrender
33
  PyOpenGL
34
  PyOpenGL-accelerate
35
  pyglet
36
+
37
+ # --- ControlNet 보조(OpenPose 등) ---
38
  controlnet-aux
39
 
40
+ # --- DINO / SAM: 반드시 Git 소스에서 설치 (PyPI 버전 없음) ---
41
+ pycocotools>=2.0.7
42
+ git+https://github.com/IDEA-Research/GroundingDINO.git@main#egg=groundingdino
43
+ git+https://github.com/facebookresearch/segment-anything.git#egg=segment-anything