diff --git a/compose/dev/docker-compose.yml b/compose/dev/docker-compose.yml
index d5a8013..71151dc 100644
--- a/compose/dev/docker-compose.yml
+++ b/compose/dev/docker-compose.yml
@@ -82,7 +82,7 @@ services:
     volumes:
       - ../../:/app
       - ../../models:/app/models
-      - hf_cache:/root/.cache/huggingface
+      - ../../hf_cache:/root/.cache/huggingface
     deploy:
       resources:
         reservations:
@@ -95,9 +95,9 @@ services:
       - WATCHFILES_FORCE_POLLING=true
       - PYTHONPATH=/app
       - HF_HOME=/root/.cache/huggingface
-      - HF_HUB_OFFLINE=1
+      - HF_HUB_OFFLINE=0
     ports:
-      - "8001:8001"
+      - "0.0.0.0:58001:8001"
     depends_on:
       fyp-redis-dev:
         condition: service_healthy
@@ -107,4 +107,3 @@ services:
 volumes:
   fyp_postgres_data:
   fyp_redis_data:
-  hf_cache:
diff --git a/compose/prod/docker-compose.inference.yml b/compose/prod/docker-compose.inference.yml
index 3a04c33..4964439 100644
--- a/compose/prod/docker-compose.inference.yml
+++ b/compose/prod/docker-compose.inference.yml
@@ -4,7 +4,6 @@ services:
     build:
       context: ../../
       dockerfile: compose/dev/inference/Dockerfile
-    restart: unless-stopped
     deploy:
       resources:
         reservations:
@@ -26,3 +25,5 @@ services:
       - '0.0.0.0:58001:8001'
     volumes:
       - ../../:/app
+      - ../../models:/app/models
+      - ../../hf_cache:/root/.cache/huggingface