Dockerfile

FROM python:3.6-slim-stretch
  
RUN apt update
RUN apt install -y python3-dev gcc

# Install pytorch and fastai
RUN pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
RUN pip install fastai

# Install starlette and uvicorn
RUN pip install starlette uvicorn python-multipart aiohttp

ADD poisonous_plants.py poisonous_plants.py
ADD Final_poisonous_plants_model.pth Final_poisonous_plants_model.pth

# Run it once to trigger resnet download
RUN python poisonous_plants.py

EXPOSE 8008

# Start the server
CMD ["python", "poisonous_plants.py", "serve"]