Add model download in dockerfile

Dilydigre · May 28, 2024 · 2d6c140 · 2d6c140
1 parent 17ed506
commit 2d6c140
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -10,6 +10,8 @@ RUN pip install --no-cache-dir -r requirements_api.txt
 
 COPY . .
 
+wget -O app/models/models_weight/ld-model.ckpt https://ommer-lab.com/files/latent-diffusion/nitro/txt2img-f8-large/model.ckpt
+
 EXPOSE 5000
 
 CMD ["uvicorn", "app.main:app", "--host","0.0.0.0", "--port", "5000"]