diff --git a/Dockerfile b/Dockerfile
index f43edef..6dd83a6 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -17,6 +17,8 @@ WORKDIR /code
 #RUN pip install --no-cache-dir -r /code/requirements.txt
 
 COPY setup.py /code/setup.py
+COPY README.md /code/README.md
+COPY src /code/src
 
 RUN pip install --no-cache-dir '.[all]'
 
@@ -53,4 +55,4 @@ EXPOSE 7860
 CMD ["python", "main.py"]
 
 #following works and gradio url works but not able to access local url on 127.0.0.1:7860
-# CMD ["python", "main.py", "--address", "0.0.0.0", "--port", "7860", "--allow-websocket-origin", "nirmits-openCHA.hf.space"]
+# CMD ["python", "main.py", "--address", "0.0.0.0", "--port", "7860", "--allow-websocket-origin", "nirmits-openCHA.hf.space"]