Update Dockerfile

Dockerfile

@@ -1,28 +1,36 @@
-# Use Python base image
-FROM python:3.9
+# Use a stable Python base image
+FROM python:3.9-slim
+
+# Set environment variables
+ENV PYTHONUNBUFFERED=1 \
+    LANG=C.UTF-8 \
+    HF_HOME="/app/huggingface_cache" \
+    HUGGINGFACE_HUB_CACHE="/app/huggingface_cache"
 
 # Set working directory
 WORKDIR /app
 
-# Copy files
+# Copy required files
 COPY requirements.txt .  
 COPY main.py .  
 
-# Install dependencies
-RUN pip install --no-cache-dir -r requirements.txt  
-
-# Set cache directory (Use HF_HOME instead of TRANSFORMERS_CACHE)
-ENV HF_HOME="/tmp/huggingface"
+# Install dependencies using a virtual environment
+RUN python -m venv /app/venv && \
+    /app/venv/bin/pip install --no-cache-dir --upgrade pip && \
+    /app/venv/bin/pip install --no-cache-dir -r requirements.txt
 
-# Ensure the directory exists
+# Ensure the model cache 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')"
+# Pre-download models (handle errors gracefully)
+RUN /app/venv/bin/python -c "from transformers import pipeline; \
+    pipeline('sentiment-analysis', model='tabularisai/multilingual-sentiment-analysis')" || echo "Failed to download model 1"
+
+RUN /app/venv/bin/python -c "from transformers import pipeline; \
+    pipeline('sentiment-analysis', model='siebert/sentiment-roberta-large-english')" || echo "Failed to download model 2"
 
 # Expose port for FastAPI
 EXPOSE 7860
 
-# Run FastAPI server
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+# Run FastAPI server using virtual environment
+CMD ["/app/venv/bin/uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]