diff --git a/.DS_Store b/.DS_Store
new file mode 100644
index 0000000..7b0093d
Binary files /dev/null and b/.DS_Store differ
diff --git a/checkpoints/best_model.pth b/checkpoints/best_model.pth
new file mode 100644
index 0000000..c23a387
Binary files /dev/null and b/checkpoints/best_model.pth differ
diff --git a/docker/.dockerignore b/docker/.dockerignore
new file mode 100644
index 0000000..760a839
--- /dev/null
+++ b/docker/.dockerignore
@@ -0,0 +1,8 @@
+.venv/
+data/
+checkpoints/
+notebooks/
+.git/
+__pycache__/
+*.pth
+*.egg-info/
\ No newline at end of file
diff --git a/docker/Dockerfile b/docker/Dockerfile
new file mode 100644
index 0000000..7fffdad
--- /dev/null
+++ b/docker/Dockerfile
@@ -0,0 +1,7 @@
+FROM python:3.11-slim
+WORKDIR /app
+COPY requirements.txt .
+RUN pip install -r requirements.txt
+COPY . .
+EXPOSE 8000
+CMD ["uvicorn", "src.inference.predict:app", "--host", "0.0.0.0", "--port", "8000"]
diff --git a/src/inference/predict.py b/src/inference/predict.py
index 8b5bd74..ba6e3d4 100644
--- a/src/inference/predict.py
+++ b/src/inference/predict.py
@@ -40,19 +40,19 @@ async def segment(request: Request,
         model = request.app.state.model
         device = request.app.state.device
 
-        with tempfile.NamedTemporaryFile(suffix=".nii", delete=False) as tmp:
+        with tempfile.NamedTemporaryFile(suffix=".nii.gz", delete=False) as tmp:
             tmp.write(await t1c.read())
             t1c_path = tmp.name
 
-        with tempfile.NamedTemporaryFile(suffix=".nii", delete=False) as tmp:
+        with tempfile.NamedTemporaryFile(suffix=".nii.gz", delete=False) as tmp:
             tmp.write(await t1n.read())
             t1n_path = tmp.name
 
-        with tempfile.NamedTemporaryFile(suffix=".nii", delete=False) as tmp:
+        with tempfile.NamedTemporaryFile(suffix=".nii.gz", delete=False) as tmp:
             tmp.write(await t2f.read())
             t2f_path = tmp.name
 
-        with tempfile.NamedTemporaryFile(suffix=".nii", delete=False) as tmp:
+        with tempfile.NamedTemporaryFile(suffix=".nii.gz", delete=False) as tmp:
             tmp.write(await t2w.read())
             t2w_path = tmp.name