diff --git a/docker-compose.yml b/docker-compose.yml
index c016597..a100312 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -1,11 +1,21 @@
 version: '3.8'
 
 services:
+  mlflow:
+    image: ghcr.io/mlflow/mlflow:v2.13.2
+    ports:
+      - "5000:5000"
+    volumes:
+      - ./mlruns:/mlruns
+    command: mlflow server --host 0.0.0.0 --port 5000 --backend-store-uri file:///mlruns
+
   api:
     build: .
     ports:
       - "8000:8000"
-    volumes:
-      - ./mlruns:/app/mlruns  # If MLflow uses local storage
+    depends_on:
+      - mlflow
     environment:
-      - MLFLOW_TRACKING_URI=http://localhost:5000  # If running MLflow server separately
\ No newline at end of file
+      - MLFLOW_TRACKING_URI=http://mlflow:5000
+    volumes:
+      - ./mlruns:/mlruns
\ No newline at end of file