Dockerfile

FROM pytorch/pytorch:latest

WORKDIR /app

COPY . .

RUN pip install -r requirements.txt

CMD [ "streamlit", "run", "app.py" ]