Update Dockerfile

Dockerfile

@@ -36,8 +36,13 @@ WORKDIR /home/user/app
 RUN pip install --no-cache-dir --upgrade pip setuptools wheel && \
     pip install --no-cache-dir -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 application port
 EXPOSE 7860
 
 # Start the Gradio app
-CMD ["conda", "run", "-n", "py312", "python", "app.py"]
+CMD ["conda", "run", "-n", "py312", "python", "app.py"]