diff --git a/Dockerfile b/Dockerfile
index 9d4c834..c4ff080 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -8,4 +8,4 @@ COPY requirements.txt /app/requirements.txt
 
 RUN pip install --no-cache-dir -r requirements.txt
 
-ENTRYPOINT ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8000"]
\ No newline at end of file
+ENTRYPOINT ["uvicorn", "ai_server:app", "--host", "0.0.0.0", "--port", "8000"]
\ No newline at end of file