dhanush1601 commited on
Commit
ee852e7
·
verified ·
1 Parent(s): 72b8dfe

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +11 -6
Dockerfile CHANGED
@@ -11,15 +11,20 @@ RUN apt-get update && apt-get install -y \
11
  tzdata \
12
  && rm -rf /var/lib/apt/lists/*
13
 
 
 
 
 
14
  RUN pip install --no-cache-dir \
15
  "torch==2.1.0" \
16
- "huggingface_hub==0.20.3" \
17
- "diffusers==0.25.0" \
18
- "transformers==4.36.0" \
19
- "accelerate==0.25.0" \
20
  sentencepiece \
21
- "imageio[ffmpeg]" \
22
- "gradio==5.23.0"
 
 
23
 
24
  COPY app.py .
25
 
 
11
  tzdata \
12
  && rm -rf /var/lib/apt/lists/*
13
 
14
+ # Install huggingface_hub first to satisfy all dependents
15
+ RUN pip install --no-cache-dir "huggingface_hub==0.28.1"
16
+
17
+ # Install ML stack compatible with hub 0.28.1
18
  RUN pip install --no-cache-dir \
19
  "torch==2.1.0" \
20
+ "diffusers==0.31.0" \
21
+ "transformers==4.45.0" \
22
+ "accelerate==0.34.0" \
 
23
  sentencepiece \
24
+ "imageio[ffmpeg]"
25
+
26
+ # Install gradio separately (also needs hub>=0.28.1)
27
+ RUN pip install --no-cache-dir "gradio==5.23.0"
28
 
29
  COPY app.py .
30