Update Dockerfile

Dockerfile

@@ -1,6 +1,9 @@
 # Read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
 # you will also find guides on how best to write your Dockerfile
 
+# Build with the command: 
+# docker build --platform linux/amd64 -t ebook2audiobook . 
+
 FROM python:3.10
 
 # Create and switch to a non-root user
@@ -24,17 +27,19 @@ RUN apt-get update && apt-get install -y git && apt-get clean && rm -rf /var/lib
 USER user
 RUN git clone https://github.com/DrewThomasson/ebook2audiobook.git /home/user/ebook2audiobook
 
-
 # Set the cloned repository as the base working directory
 WORKDIR /home/user/ebook2audiobook
-RUN git checkout viet_bandaid
 
 #Install Python dependences from the ebook2audiobook repo
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
+# Do a test run to make sure that the base models are pre-downloaded and baked into the image
+# RUN echo "This is a test sentence." > test.txt 
+# RUN python app.py --headless --ebook test.txt
+# RUN rm test.txt
+
 # Expose the required port
 EXPOSE 7860
 
 # Start the Gradio app from the repository
-CMD ["python", "app.py"]
-
+CMD ["python", "app.py"]