Dockerfile

FROM bitnami/pytorch

WORKDIR /app

COPY ./requirements.txt /app/requirements.txt

RUN pip install --no-cache-dir -r /app/requirements.txt

COPY . /app

EXPOSE 7860
CMD ["streamlit", "run", "/app/app.py"]