thejagstudio commited on
Commit
7c86984
·
verified ·
1 Parent(s): 411eadc

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -0
Dockerfile CHANGED
@@ -1,5 +1,9 @@
1
  FROM python:3
2
  WORKDIR /data/app
 
 
 
 
3
  COPY requirements.txt ./
4
  RUN pip install -r requirements.txt
5
  COPY . .
 
1
  FROM python:3
2
  WORKDIR /data/app
3
+ # Download the model file
4
+ RUN apt-get update && apt-get install -y wget \
5
+ && wget -O /data/app/sneaker_category_predictor_v2.h5 "https://huggingface.co/thejagstudio/SneakerAI/resolve/main/sneaker_category_predictor_v2.h5?download=true" \
6
+ && apt-get clean
7
  COPY requirements.txt ./
8
  RUN pip install -r requirements.txt
9
  COPY . .