diff --git a/Dockerfile b/Dockerfile
index 45f7fdd..d2b831d 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -5,9 +5,10 @@ WORKDIR /app
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-COPY agent-image-t2i.py .
+COPY main.py .
+COPY utils/ ./utils/
 RUN mkdir -p /app/output
 
 EXPOSE 8000
 
-CMD ["uvicorn", "agent-image-t2i:app", "--host", "0.0.0.0", "--port", "8000"]
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]