hdremover commited on
Commit
844da06
·
verified ·
1 Parent(s): 5ac5cb9

Create requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +28 -0
requirements.txt ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # CPU-only torch (saves 1.5GB vs CUDA wheel)
2
+ --extra-index-url https://download.pytorch.org/whl/cpu
3
+ torch==2.7.1+cpu
4
+ torchvision==0.22.1+cpu
5
+
6
+ # Core
7
+ transformers>=4.40.0
8
+ Pillow>=10.0.0
9
+ numpy>=1.24.0
10
+ gradio>=4.31.0
11
+ fastapi>=0.110.0
12
+ uvicorn>=0.27.0
13
+ python-multipart>=0.0.9
14
+ requests>=2.31.0
15
+ huggingface_hub>=0.23.0
16
+
17
+ # Models
18
+ # rembg dropped Python 3.13 support — install direct from GitHub
19
+ rembg @ git+https://github.com/danielgatis/rembg.git
20
+ # ben2 not on PyPI — install from HF repo source
21
+ git+https://github.com/PramaLLC/BEN2.git
22
+ onnxruntime>=1.17.0
23
+
24
+ # BiRefNet / RMBG-2.0 deps
25
+ einops>=0.7.0
26
+ kornia>=0.7.3
27
+ timm>=0.9.12
28
+ opencv-contrib-python-headless>=4.9.0