File size: 460 Bytes
9251269
a825d34
 
 
 
 
 
 
 
9251269
a825d34
 
9251269
 
 
 
 
a825d34
 
 
 
9251269
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
FROM python:3.11-slim

WORKDIR /app

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

COPY requirements.txt .
RUN pip3 install -r requirements.txt

COPY app.py .
COPY model.py .
COPY gen_a2b_fp16.pth .
COPY gen_b2a_fp16.pth .

EXPOSE 8501

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

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