Update Dockerfile

Dockerfile

@@ -5,7 +5,9 @@ FROM python:3.9
 
 WORKDIR /code
 
-COPY ./main.py /code/main.py
+COPY ./serve.py /code/serve.py
+
+COPY ./cerebras.py /code/cerebras.py
 
 COPY ./space_checker.py /code/space_checker.py