Dockerfile

FROM tensorflow/tensorflow:2.12.0
# Install Dependencies
RUN apt-get update -q \
 && apt-get install -qy --no-install-recommends wget git libopencv-dev python3-opencv \
 && apt-get autoremove -y \
 && apt-get clean \
 && rm -rf /var/lib/apt/lists/*

RUN mkdir -p /.cache && chmod -R 777 /.cache

WORKDIR /data/app

COPY requirements.txt ./
RUN pip install -r requirements.txt
# Download the model file
# RUN wget -O /data/app/sneaker_category_predictor_v2.h5 "https://huggingface.co/thejagstudio/SneakerAI/resolve/main/sneaker_category_predictor_v2.h5?download=true"
COPY . .
EXPOSE 7860
CMD ["python","/data/app/app.py"]