Update Dockerfile
Browse files- Dockerfile +6 -2
Dockerfile
CHANGED
@@ -3,9 +3,9 @@ FROM python:3.11-slim as builder
|
|
3 |
|
4 |
WORKDIR /app
|
5 |
|
6 |
-
# Install git
|
7 |
RUN apt-get update && \
|
8 |
-
apt-get install -y --no-install-recommends git && \
|
9 |
apt-get clean && \
|
10 |
rm -rf /var/lib/apt/lists/*
|
11 |
|
@@ -15,6 +15,10 @@ RUN git clone -b patch-1 https://github.com/zawsq/HeartDiseasePredictor_Model.gi
|
|
15 |
# Set the working directory to the cloned repository
|
16 |
WORKDIR /app/HeartDiseasePredictor_Model
|
17 |
|
|
|
|
|
|
|
|
|
18 |
# Set PYTHONPATH to include the root directory of the project
|
19 |
ENV PYTHONPATH=/app/HeartDiseasePredictor_Model
|
20 |
|
|
|
3 |
|
4 |
WORKDIR /app
|
5 |
|
6 |
+
# Install git and wget
|
7 |
RUN apt-get update && \
|
8 |
+
apt-get install -y --no-install-recommends git wget && \
|
9 |
apt-get clean && \
|
10 |
rm -rf /var/lib/apt/lists/*
|
11 |
|
|
|
15 |
# Set the working directory to the cloned repository
|
16 |
WORKDIR /app/HeartDiseasePredictor_Model
|
17 |
|
18 |
+
# Download the model file from Hugging Face into the existing "model" folder
|
19 |
+
RUN wget -O /app/HeartDiseasePredictor_Model/model/RandomForest_best_model.pkl \
|
20 |
+
https://huggingface.co/exis0705/heart_disease_predictor_model/resolve/main/RandomForest_best_model.pkl
|
21 |
+
|
22 |
# Set PYTHONPATH to include the root directory of the project
|
23 |
ENV PYTHONPATH=/app/HeartDiseasePredictor_Model
|
24 |
|