samwaugh commited on
Commit
944a1aa
·
1 Parent(s): ee7d732

Phase 2A.2: Add ML dependencies for CLIP/PaintingCLIP inference

Browse files
Files changed (1) hide show
  1. requirements.txt +16 -2
requirements.txt CHANGED
@@ -1,6 +1,20 @@
 
1
  flask>=3
2
  gunicorn>=21
3
  flask-cors
 
 
4
  huggingface_hub>=0.20
5
- pillow
6
- numpy
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Web framework and server
2
  flask>=3
3
  gunicorn>=21
4
  flask-cors
5
+
6
+ # Hugging Face ecosystem
7
  huggingface_hub>=0.20
8
+
9
+ # Core ML libraries
10
+ torch>=2.0.0
11
+ transformers>=4.30.0
12
+ peft>=0.4.0
13
+
14
+ # Computer vision and image processing
15
+ pillow>=10.0.0
16
+ opencv-python>=4.8.0
17
+ numpy>=1.24.0
18
+
19
+ # Optional: GPU acceleration (if available)
20
+ # torchvision>=0.15.0 # Uncomment if you need additional vision models