jbilcke-hf HF staff commited on
Commit
6525751
โ€ข
1 Parent(s): af08fb2

remove bloat in deps + add avif

Browse files
Files changed (3) hide show
  1. Dockerfile +0 -7
  2. app.py +2 -0
  3. requirements.txt +7 -29
Dockerfile CHANGED
@@ -49,13 +49,6 @@ RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
49
  #RUN bun i
50
  #RUN bun build ./src/index.tsx --outdir ../public/
51
 
52
- RUN echo "Installing openmim and mim dependencies"
53
- RUN pip3 install --no-cache-dir -U openmim
54
- RUN mim install mmengine
55
- RUN mim install "mmcv>=2.0.1"
56
- RUN mim install "mmdet>=3.3.0"
57
- RUN mim install "mmpose>=1.3.2"
58
-
59
  WORKDIR $HOME/app
60
 
61
  COPY --chown=user . $HOME/app
 
49
  #RUN bun i
50
  #RUN bun build ./src/index.tsx --outdir ../public/
51
 
 
 
 
 
 
 
 
52
  WORKDIR $HOME/app
53
 
54
  COPY --chown=user . $HOME/app
app.py CHANGED
@@ -16,7 +16,9 @@ import signal
16
  from typing import Dict, Any, List, Optional
17
  import base64
18
  import io
 
19
  from PIL import Image
 
20
 
21
  # Configure logging
22
  logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
 
16
  from typing import Dict, Any, List, Optional
17
  import base64
18
  import io
19
+
20
  from PIL import Image
21
+ import pillow_avif
22
 
23
  # Configure logging
24
  logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
requirements.txt CHANGED
@@ -16,10 +16,6 @@ torchaudio
16
  # note: gradio is only used for the cropping utility
17
  gradio==5.0.0b4
18
 
19
- gradio-webrtc==0.0.1
20
-
21
- opuslib==3.0.1
22
-
23
  pyyaml==6.0.1
24
  numpy==1.22.0 # Updated to resolve conflicts
25
  opencv-python==4.8.1.78 # Downgraded to be compatible with numpy
@@ -39,8 +35,8 @@ ffmpeg-python==0.2.0
39
  onnxruntime-gpu==1.19.2
40
  onnx==1.16.2
41
  scikit-image==0.20.0
42
- albumentations==1.3.1
43
- matplotlib==3.7.2
44
  tyro==0.8.5
45
  chumpy==0.70
46
 
@@ -48,40 +44,22 @@ diffusers==0.30.3
48
  accelerate==0.34.2
49
  tensorflow==2.12.0
50
  tensorboard==2.12.0
51
- soundfile==0.12.1
52
  transformers==4.39.2
 
 
53
 
54
  gdown==5.2.0
55
  requests==2.32.3
56
  omegaconf==2.3.0
57
- moviepy==1.0.3
58
-
59
  pydantic==2.9.2
60
-
61
- # --------------------------------------------------------------------
62
- # RESERVED FOR FUTURE USAGE
63
- #
64
- # (it adds bloat, so you can remove them if you want)
65
- # --------------------------------------------------------------------
66
  dnspython>=2.0.0
67
  ifaddr>=0.2.0
68
- aioice==0.9.0
69
- aiortc==1.9.0
70
  aiohttp==3.10.5
71
- av==12.3.0
72
- einops==0.7.0
73
- safetensors==0.4.5
74
- sentencepiece==0.2.0
75
- sounddevice==0.5.0
76
- sphn==0.1.4
77
- huggingface-hub==0.25.1
78
- optimum-quanto==0.2.4
79
- httpx==0.24.1
80
- timm==1.0.9
81
- ftfy==6.2.3
82
- facexlib==0.3.0
83
 
84
  # --------------------------------------------------------------------
85
  # Used for advanced LivePortrait features
86
  # --------------------------------------------------------------------
87
  pillow==10.4.0
 
 
 
 
16
  # note: gradio is only used for the cropping utility
17
  gradio==5.0.0b4
18
 
 
 
 
 
19
  pyyaml==6.0.1
20
  numpy==1.22.0 # Updated to resolve conflicts
21
  opencv-python==4.8.1.78 # Downgraded to be compatible with numpy
 
35
  onnxruntime-gpu==1.19.2
36
  onnx==1.16.2
37
  scikit-image==0.20.0
38
+ albumentations==1.4.10
39
+ matplotlib==3.9.0
40
  tyro==0.8.5
41
  chumpy==0.70
42
 
 
44
  accelerate==0.34.2
45
  tensorflow==2.12.0
46
  tensorboard==2.12.0
 
47
  transformers==4.39.2
48
+ huggingface-hub==0.25.1
49
+ safetensors==0.4.5
50
 
51
  gdown==5.2.0
52
  requests==2.32.3
53
  omegaconf==2.3.0
 
 
54
  pydantic==2.9.2
 
 
 
 
 
 
55
  dnspython>=2.0.0
56
  ifaddr>=0.2.0
 
 
57
  aiohttp==3.10.5
 
 
 
 
 
 
 
 
 
 
 
 
58
 
59
  # --------------------------------------------------------------------
60
  # Used for advanced LivePortrait features
61
  # --------------------------------------------------------------------
62
  pillow==10.4.0
63
+
64
+ # by popular demand, let's add AVIF
65
+ pillow-avif-plugin==1.4.6