pixelsdesign commited on
Commit
f89d019
·
verified ·
1 Parent(s): bdf55d1

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +20 -7
requirements.txt CHANGED
@@ -1,17 +1,30 @@
1
- torch
 
 
 
 
 
2
  accelerate
 
 
3
  opencv-python
4
- spaces
5
  pillow
6
  numpy
7
  timm
8
  kornia
9
- prettytable
10
- typing
11
  scikit-image
 
 
 
 
12
  huggingface_hub
13
  transformers>=4.39.1
14
- gradio
 
 
 
15
  gradio_imageslider
16
- loadimg>=0.1.1
17
- einops
 
 
 
1
+ # CPU-only PyTorch
2
+ torch==2.1.0+cpu
3
+ torchvision==0.16.0+cpu
4
+ torchaudio==2.1.0+cpu
5
+
6
+ # Accelerate for Hugging Face
7
  accelerate
8
+
9
+ # Image and vision libraries
10
  opencv-python
 
11
  pillow
12
  numpy
13
  timm
14
  kornia
 
 
15
  scikit-image
16
+ loadimg>=0.1.1
17
+ einops
18
+
19
+ # Hugging Face and Spaces
20
  huggingface_hub
21
  transformers>=4.39.1
22
+ spaces
23
+
24
+ # Gradio UI
25
+ gradio==3.39.0
26
  gradio_imageslider
27
+
28
+ # Utilities
29
+ prettytable
30
+ typing