Spaces:
Sleeping
Sleeping
File size: 969 Bytes
1e4a524 33a72a2 1e4a524 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
SHELL ["/bin/bash", "-c"]
# Environment variables for Hugging Face cache
ENV HF_HOME=/app/hf_cache
ENV TRANSFORMERS_CACHE=/app/hf_cache
ENV HF_TOKEN=${HF_TOKEN}
ENV PATH=/opt/conda/bin:$PATH
ENV MPLCONFIGDIR=/tmp/matplotlib
# Install system dependencies
RUN apt-get update && apt-get install -y \
git wget curl unzip ffmpeg libgl1-mesa-glx libglib2.0-0 && \
apt-get clean
# Set up working directory as /app
WORKDIR /app
# Copy project into /app
COPY . /app
# Create conda environment and install dependencies
COPY requirements.txt /app/requirements.txt
RUN conda create -n epic python=3.10 -y && \
conda run -n epic pip install --upgrade pip && \
conda run -n epic pip install -r /app/requirements.txt
RUN chmod -R 777 /app /workspace
# Expose Gradio default port
EXPOSE 7860
# Start the Gradio app
CMD ["conda", "run", "--no-capture-output", "-n", "epic", "python", "track-pixels_gradio.py"]
|