Update Dockerfile
Browse files- Dockerfile +6 -1
Dockerfile
CHANGED
|
@@ -14,7 +14,12 @@ RUN chmod -R 777 /tmp
|
|
| 14 |
COPY requirements.txt .
|
| 15 |
RUN pip install --no-cache-dir -r requirements.txt
|
| 16 |
|
| 17 |
-
#
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
COPY app/ ./app/
|
| 19 |
|
| 20 |
# Ensure app directory is readable
|
|
|
|
| 14 |
COPY requirements.txt .
|
| 15 |
RUN pip install --no-cache-dir -r requirements.txt
|
| 16 |
|
| 17 |
+
# Download the model during the build process
|
| 18 |
+
RUN mkdir -p /app/model && \
|
| 19 |
+
python3 -c "from huggingface_hub import hf_hub_download; \
|
| 20 |
+
hf_hub_download(repo_id='Miguel764/efficientnetv2s-skin-cancer-classifier', \
|
| 21 |
+
filename='efficientnetv2s.h5', local_dir='/app/model')"
|
| 22 |
+
|
| 23 |
COPY app/ ./app/
|
| 24 |
|
| 25 |
# Ensure app directory is readable
|