SilentWraith commited on
Commit
ff402b2
·
verified ·
1 Parent(s): c473a31

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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