File size: 974 Bytes
06031b4
ac1a7b9
06031b4
ac1a7b9
cd51489
 
 
d33be4e
cd51489
 
0bc7d4b
d2d0ccd
 
06031b4
 
cd51489
 
 
 
 
619cd00
 
 
f3a3457
ac1a7b9
 
 
d33be4e
 
ac1a7b9
3144c3a
723e376
8ab500d
cd51489
8ab500d
d43322e
d2d0ccd
ac1a7b9
cd51489
ac1a7b9
cd51489
ac1a7b9
a3cdc31
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
37
38
39
40
41
42
43
FROM python:3.10-slim

WORKDIR /app

ENV HF_HOME=/tmp/huggingface
ENV TRANSFORMERS_CACHE=/tmp/huggingface/transformers
ENV HF_DATASETS_CACHE=/tmp/huggingface/datasets

ENV PYTHONDONTWRITEBYTECODE=1
ENV PYTHONUNBUFFERED=1
ENV CUDA_VISIBLE_DEVICES=""
ENV HF_TOKEN=${HF_TOKEN}

RUN apt-get update && apt-get install -y \
    build-essential \
    libgl1-mesa-glx \
    libglib2.0-0 \
    libsm6 \
    libxext6 \
    libxrender-dev \
    cmake \
    libopenblas-dev \
    liblapack-dev \
    libopencv-dev \
    && apt-get clean \
    && rm -rf /var/lib/apt/lists/*

RUN mkdir -p /tmp/uploads /tmp/results /tmp/huggingface/transformers /tmp/huggingface/datasets  \
    && chmod -R 777 /tmp/uploads /tmp/results /tmp/huggingface 



# Copy requirements file and install dependencies
COPY requirements.txt .
RUN pip install --no-cache-dir -U pip && \
    pip install --no-cache-dir -r requirements.txt && \
    pip cache purge

COPY app.py .

EXPOSE 7860

CMD ["python", "app.py"]