diff --git a/README.md b/README.md
index 6c24219..e68c008 100644
--- a/README.md
+++ b/README.md
@@ -30,21 +30,14 @@
 
 2. Run `install.sh` to download and extract models:
     ```sh
-    mkdir -p data/models
-    mkdir -p data/tmp
-    mkdir -p data/predictions/thumbnails
-    mkdir -p data/media
-    mkdir -p data/analyser
-    mkdir -p data/bentoml
-    mkdir -p data/backend_cache
-    wget https://tib.eu/cloud/s/2z3ocpRaMbjnRrF/download/models.tar.gz
+    mkdir data/cache
+    mkdir data/analyser
+    mkdir data/media
+    mkdir data/tmp
+    mkdir data/predictions
+    mkdir data/backend_cache
+    wget wget https://tib.eu/cloud/s/xLXEEsf99KPYcoW/download/models.tar.gz
     tar -xf models.tar.gz --directory data/
-    # CPU only
-    wget https://tib.eu/cloud/s/MnkZjS87DZKa79Z/download/bentoml.tar.gz
-    tar -xf bentoml.tar.gz --directory data/
-    # GPU only
-    wget https://tib.eu/cloud/s/EnQB67NsN7Y7MFX/download/bentoml_cuda.tar.gz
-    tar -xf bentoml_cuda.tar.gz --directory data/bentoml
     ```
 
 3. Build and start the container:
diff --git a/docker-compose.yml b/docker-compose.yml
index e9c0437..0230a2d 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -116,6 +116,11 @@ services:
         source: ./data/tmp
         target: /tmp
 
+    depends_on:
+      - postgres
+      
+    restart: always
+
     working_dir: /app/backend/
     environment:
       - PYTHONPATH=/app:/app/backend/
@@ -161,6 +166,7 @@ services:
       - type: bind
         source: ./data/tmp
         target: /tmp
+        
 
     working_dir: /app/backend/
     environment:
@@ -171,6 +177,9 @@ services:
     depends_on:
       - postgres
       - memcached
+      - backend_celery
+
+    restart: always
 
   frontend:
     image: frontend