Update Dockerfile

Dockerfile

@@ -17,7 +17,7 @@ RUN apt-get update && apt-get install -y \
     build-essential \
     && rm -rf /var/lib/apt/lists/*
 
-# 5. Create a non-root user (Hugging Face Requirement)
+# 5. Create a non-root user (Hugging Face requirement)
 RUN useradd -m -u 1000 user
 RUN mkdir -p /code/.cache && chown -R user:user /code
 
@@ -31,8 +31,9 @@ RUN pip install --no-cache-dir --user --upgrade -r /code/requirements.txt
 # 8. Copy the entire project
 COPY --chown=user . /code
 
-# 9. Expose port 7860
-EXPOSE 7860
+# 9. Pre-download models during Build
+RUN python -m app.setup_models
 
 # 10. Start the application
+EXPOSE 7860
 CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]