test / Dockerfile
JaynilJaiswal's picture
Update Dockerfile
25c46f8 verified
raw
history blame contribute delete
1.22 kB
# Use a base image (Python 3.10 slim)
FROM python:3.10-slim
ENV HF_HOME=/data
# Set working directory
WORKDIR /code
# Copy requirements first to leverage Docker cache
COPY ./requirements.txt /code/requirements.txt
# Install system dependencies (curl for downloading), Python dependencies,
# create /data dir, check for model and download if missing, then clean up.
RUN apt-get update && \
apt-get install -y --no-install-recommends curl && \
pip install --no-cache-dir --upgrade pip && \
pip install --no-cache-dir -r /code/requirements.txt
# RUN echo "Checking for model file in /data..." && \
# if [ ! -f /data/zephyr-7b-beta.Q4_K_M.gguf ]; then \
# echo "Model file not found. Downloading..." && \
# curl -L -o /data/zephyr-7b-beta.Q4_K_M.gguf https://huggingface.co/TheBloke/zephyr-7B-beta-GGUF/resolve/main/zephyr-7b-beta.Q4_K_M.gguf && \
# echo "Download complete."; \
# else \
# echo "Model file already exists."; \
# fi
# Copy the rest of the application code from the build context to the container
COPY . /code/
# Set the command to run the Gradio app (app.py)
# HF Spaces usually figures this out, but explicitly setting it is fine.
CMD ["python", "app.py"]