Update Dockerfile

Dockerfile

@@ -11,8 +11,14 @@ COPY main.py .
 # Install dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Set cache directory
+ENV TRANSFORMERS_CACHE="/app/cache"
+
+# Pre-download the model (to avoid runtime downloads)
+RUN python -c "from transformers import pipeline; pipeline('sentiment-analysis', model='distilbert-base-uncased-finetuned-sst-2-english')"
+
 # Expose port for FastAPI
 EXPOSE 7860
 
 # Run FastAPI server
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]