Dockerfile

FROM python:3.9

WORKDIR /data/app

RUN apt-get update && apt-get install -y libgl1 && apt-get clean


RUN pip install --upgrade pip && pip install tensorflow

COPY requirements.txt ./
RUN pip install -r requirements.txt

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/main.py"]