diff --git a/ai-search-demo/Dockerfile b/ai-search-demo/Dockerfile
index 29522f3..38f5ca3 100644
--- a/ai-search-demo/Dockerfile
+++ b/ai-search-demo/Dockerfile
@@ -9,6 +9,13 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     git \
     git-lfs
 
+WORKDIR /app
+
 RUN pip install --upgrade pip
 COPY requirements.txt requirements.txt
 RUN pip install -r requirements.txt
+
+COPY . .
+ENV PYTHONPATH /app/
+
+CMD streamlit run --server.port 8000 --server.address 0.0.0.0 ai_search_demo/ui.py