diff --git a/Dockerfile b/Dockerfile
index be0eeb9..fd55f68 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -17,4 +17,4 @@ RUN pip install --no-cache-dir -r requirements.txt
 COPY . .
 
 # Run the application
-CMD ["python", "./app.py"]
\ No newline at end of file
+CMD ["python3", "-m", "fastapi", "run", "--host", "0.0.0.0", "--port", "8000"]
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
index 575cd10..84f45cc 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,4 +4,5 @@ torch
 torchvision
 ultralytics
 opencv-python
-websockets
\ No newline at end of file
+websockets
+fastapi[standard]
\ No newline at end of file