Update Dockerfile

Dockerfile

@@ -1,6 +1,6 @@
 FROM python:3.9-slim
 
-# Install FFmpeg (Required for Video Downloading and Audio Processing)
+# Install FFmpeg for Voice/Video
 RUN apt-get update && apt-get install -y ffmpeg git && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /app
@@ -10,10 +10,12 @@ ENV XDG_CACHE_HOME=/tmp
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Pre-download the Whisper AI Voice model during build so it's instantly ready
+# Pre-download the Voice Model
 RUN python -c "import whisper; whisper.load_model('base')"
 
-# Copy the app code and start the API
+# Pre-download the Uncensored Image Model
+RUN python -c "from diffusers import StableDiffusionPipeline; StableDiffusionPipeline.from_pretrained('prompthero/openjourney', safety_checker=None)"
+
 COPY app.py .
 EXPOSE 7860
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]