File size: 958 Bytes
628a35d f932697 628a35d 30bfb87 628a35d f932697 30bfb87 |
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 |
FROM registry.hf.space/microsoft-omniparser:latest
USER root
# Set proper permissions for /tmp
RUN chmod 1777 /tmp
# Install CUDA dependencies and system packages
RUN apt update -q && apt install -y --no-install-recommends \
ca-certificates wget libgl1 \
&& wget -qO /tmp/cuda-keyring.deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb \
&& dpkg -i /tmp/cuda-keyring.deb && apt update -q \
&& apt install -y --no-install-recommends libcudnn8 libcublas-12-2 \
&& rm -rf /var/lib/apt/lists/* /tmp/cuda-keyring.deb
# Install Python dependencies
COPY requirements.txt requirements.txt
RUN pip install --no-cache-dir -r requirements.txt
# Copy application files
COPY main.py main.py
COPY weights.py weights.py
# Expose application port
EXPOSE 7860
# Run weights.py and start the FastAPI server
CMD ["bash", "-c", "python weights.py && uvicorn main:app --host 0.0.0.0 --port 7860"]
|