dschandra commited on
Commit
39ccc90
·
verified ·
1 Parent(s): 2c45196

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +8 -27
requirements.txt CHANGED
@@ -1,27 +1,8 @@
1
- # Web Framework
2
- Flask==2.3.2
3
-
4
- # Deep Learning Libraries
5
- torch==2.0.1
6
- torchvision==0.15.1
7
-
8
- # Image Segmentation and Object Detection Models
9
- segmentation-models-pytorch==0.3.0
10
- ultralytics==8.0.1 # For YOLOv5, which can be used with segmentation
11
-
12
- # Image Processing
13
- Pillow==9.5.0
14
- opencv-python==4.8.0.74
15
-
16
- # Diffusion Models (Optional - if you decide to use them)
17
- diffusers==0.19.3
18
-
19
- # Utility Libraries
20
- numpy==1.24.3
21
- scipy==1.11.3
22
-
23
- # JSON Handling
24
- Flask-JSON==0.3.4
25
-
26
- # For base64 image conversion
27
- base64==1.0.3
 
1
+ accelerate
2
+ diffusers
3
+ invisible_watermark
4
+ torch
5
+ transformers
6
+ xformers
7
+ opencv-python
8
+ requests