Steveeeeeeen HF staff commited on
Commit
36f27fe
·
verified ·
1 Parent(s): 68a291c

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +8 -8
Dockerfile CHANGED
@@ -45,20 +45,20 @@ WORKDIR /app
45
  # Clone the repository
46
  RUN git clone https://github.com/Zyphra/Zonos.git && cd Zonos
47
 
48
- # Copy application files
49
- WORKDIR /app/Zonos
50
-
51
- # Set environment variable for Triton cache
52
  ENV TRITON_CACHE_DIR=/tmp/.triton
 
53
 
54
- # Ensure Triton cache directory is writable
55
- RUN mkdir -p $TRITON_CACHE_DIR && chmod -R 777 $TRITON_CACHE_DIR
56
 
57
  # Install Python dependencies
 
58
  RUN uv pip install --system -e . && uv pip install --system -e .[compile]
59
 
60
  # Expose the Gradio default port
61
  EXPOSE 7860
62
 
63
- # Run the Gradio app
64
- CMD ["python", "gradio_interface.py"]
 
 
45
  # Clone the repository
46
  RUN git clone https://github.com/Zyphra/Zonos.git && cd Zonos
47
 
48
+ # Set environment variables for writable cache directories
 
 
 
49
  ENV TRITON_CACHE_DIR=/tmp/.triton
50
+ ENV TRANSFORMERS_CACHE=/tmp/huggingface_cache
51
 
52
+ # Ensure cache directories are writable
53
+ RUN mkdir -p $TRITON_CACHE_DIR $TRANSFORMERS_CACHE && chmod -R 777 $TRITON_CACHE_DIR $TRANSFORMERS_CACHE
54
 
55
  # Install Python dependencies
56
+ WORKDIR /app/Zonos
57
  RUN uv pip install --system -e . && uv pip install --system -e .[compile]
58
 
59
  # Expose the Gradio default port
60
  EXPOSE 7860
61
 
62
+ # Run the Gradio app from /app
63
+ WORKDIR /app
64
+ CMD ["python", "app.py"]