diff --git a/Dockerfile b/Dockerfile
index cae3ec1..570e682 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,9 +1,10 @@
-FROM python:3.10
+FROM python:3.10-alpine
 WORKDIR /app
 
 COPY requirements_api.txt .
 COPY requirements_ai.txt .
 
+RUN apk update && apk add git wget cargo
 RUN pip3 install --upgrade pip
 RUN pip install --no-cache-dir --no-deps -r requirements_ai.txt
 RUN pip install --no-cache-dir -r requirements_api.txt
@@ -12,6 +13,20 @@ COPY . .
 
 RUN wget -q -O app/models/models_weight/ld-model.ckpt https://ommer-lab.com/files/latent-diffusion/nitro/txt2img-f8-large/model.ckpt
 
+RUN mkdir repos
+WORKDIR repos
+
+RUN curl -s https://packagecloud.io/install/repositories/github/git-lfs/script.deb.sh | bash
+RUN apk install git-lfs
+RUN git lfs install
+RUN git clone https://github.com/Dilydigre/ai.git
+
+WORKDIR ../
+
+RUN cp repos/ai/app/models/models_weight/modele1.ckpt app/models/models_weight/modele1.ckpt
+RUN rm -rf repos
+
+
 EXPOSE 5000
 
 CMD ["uvicorn", "app.main:app", "--host","0.0.0.0", "--port", "5000"]
\ No newline at end of file
diff --git a/requirements_ai.txt b/requirements_ai.txt
index 4b6bcaa..b978a6e 100644
--- a/requirements_ai.txt
+++ b/requirements_ai.txt
@@ -62,7 +62,7 @@ multidict==6.0.5
 namex==0.0.8
 networkx==3.3
 numpy==1.26.4
-nvidia-cublas-cu12==12.1.3.1
+nvidia-cublas-cu12
 nvidia-cuda-cupti-cu12==12.1.105
 nvidia-cuda-nvrtc-cu12==12.1.105
 nvidia-cuda-runtime-cu12==12.1.105