Update Dockerfile

Dockerfile

@@ -5,17 +5,21 @@ FROM python:3.9
 WORKDIR /app
 
 # Copy files
-COPY requirements.txt .
-COPY main.py .
+COPY requirements.txt .  
+COPY main.py .  
 
 # Install dependencies
-RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt  
 
-# Set cache directory
-ENV TRANSFORMERS_CACHE="/app/cache"
+# Set cache directory (Use HF_HOME instead of TRANSFORMERS_CACHE)
+ENV HF_HOME="/tmp/huggingface"
 
-# 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')"
+# Ensure the directory exists
+RUN mkdir -p $HF_HOME
+
+# Pre-download the correct models (Multilingual + English sentiment analysis)
+RUN python -c "from transformers import pipeline; pipeline('sentiment-analysis', model='tabularisai/multilingual-sentiment-analysis')"
+RUN python -c "from transformers import pipeline; pipeline('sentiment-analysis', model='siebert/sentiment-roberta-large-english')"
 
 # Expose port for FastAPI
 EXPOSE 7860