gdleds commited on
Commit
edd67b4
·
1 Parent(s): 3b026a5
Files changed (2) hide show
  1. Dockerfile +32 -0
  2. requirements.txt +7 -0
Dockerfile ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM python:3.9-slim
2
+
3
+ WORKDIR /home/app
4
+
5
+ # Installation des dépendances système
6
+ RUN apt-get update && apt-get install -y \
7
+ build-essential \
8
+ libpq-dev \
9
+ git \
10
+ && rm -rf /var/lib/apt/lists/*
11
+
12
+ # Installation des dépendances Python
13
+ COPY requirements.txt requirements.txt
14
+ RUN pip install -r requirements.txt
15
+
16
+
17
+
18
+ # Exposition du port MLflow
19
+ EXPOSE 5000
20
+
21
+ # Définir les variables d'environnement pour AWS et MLflow
22
+ ENV AWS_ACCESS_KEY_ID=$AWS_ACCESS_KEY_ID
23
+ ENV AWS_SECRET_ACCESS_KEY=$AWS_SECRET_ACCESS_KEY
24
+ ENV BACKEND_STORE_URI=$BACKEND_STORE_URI
25
+ ENV MLFLOW_DEFAULT_ARTIFACT_ROOT=$MLFLOW_DEFAULT_ARTIFACT_ROOT
26
+
27
+
28
+ # Commande pour démarrer le serveur MLflow
29
+ CMD mlflow server -p $PORT \
30
+ --host 0.0.0.0 \
31
+ --backend-store-uri $BACKEND_STORE_URI \
32
+ --default-artifact-root $MLFLOW_DEFAULT_ARTIFACT_ROOT
requirements.txt ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ mlflow==2.15.0
2
+ psycopg2-binary
3
+ boto3
4
+ sqlalchemy
5
+ scikit-learn
6
+ pandas
7
+ numpy