File size: 753 Bytes
6c0250f
e13cd9e
 
 
 
 
 
 
 
 
 
 
 
 
b030c9f
 
e13cd9e
e0bc8d0
 
 
e13cd9e
 
 
 
 
3509591
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
FROM python:3.10.11

WORKDIR /app

RUN apt-get update && apt-get install -y \
    build-essential \
    curl \
    software-properties-common \
    git \
    && rm -rf /var/lib/apt/lists/*

COPY requirements.txt ./
COPY src/ ./src/

RUN pip3 uninstall -y torch torchaudio torchvision

RUN pip3 install -r requirements.txt
RUN pip3 install torch==2.7.0 --index-url https://download.pytorch.org/whl/cu126
RUN pip3 install torchaudio==2.7.0 --index-url https://download.pytorch.org/whl/cu126
RUN pip3 install torchvision==0.22.0 --index-url https://download.pytorch.org/whl/cu126

EXPOSE 8501

HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health

ENTRYPOINT ["streamlit", "run", "src/app.py", "--server.port=8501", "--server.address=0.0.0.0"]