Update Dockerfile

Dockerfile

@@ -1,46 +1,44 @@
-# Use a stable Python base image
-FROM python:3.9-slim
+# Use the official Python 3.9 slim image
+FROM python:3.9-slim AS base
 
 # Set environment variables
 ENV PYTHONUNBUFFERED=1 \
     LANG=C.UTF-8 \
-    HF_HOME="/app/huggingface_cache" \
-    HUGGINGFACE_HUB_CACHE="/app/huggingface_cache"
+    HF_HOME="/tmp/huggingface_cache" \
+    HUGGINGFACE_HUB_CACHE="/tmp/huggingface_cache"
 
 # Set working directory
 WORKDIR /app
 
-# Copy the required files
+# Copy requirements file
 COPY requirements.txt .  
-COPY main.py .  
 
 # Install dependencies using a virtual environment
+RUN apt-get update && apt-get install -y curl && rm -rf /var/lib/apt/lists/*
+
 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 model cache directory exists
-RUN mkdir -p $HF_HOME
+# Ensure the Hugging Face cache directory exists
+RUN mkdir -p $HF_HOME && chmod -R 777 $HF_HOME
+
+# Copy application files
+COPY main.py .  
 
-# Add Hugging Face token as an environment variable in the container (this will be injected by Hugging Face Spaces' secrets management)
+# Set up Hugging Face token as a build argument (secured via environment variables)
 ARG HF_TOKEN
 ENV HF_TOKEN=${HF_TOKEN}
 
-ENV HF_HOME="/tmp/huggingface_cache"
-ENV HUGGINGFACE_HUB_CACHE="/tmp/huggingface_cache"
-
-RUN mkdir -p $HF_HOME && chmod -R 777 $HF_HOME
-
-
-# Pre-download models (handle errors gracefully)
+# Pre-download models
 RUN /app/venv/bin/python -c "from transformers import pipeline; \
-    pipeline('sentiment-analysis', model='tabularisai/multilingual-sentiment-analysis', use_auth_token='$HF_TOKEN')" || echo 'Failed to download model 1'
+    pipeline('sentiment-analysis', model='Ehrii/sentiment', use_auth_token='$HF_TOKEN')" || echo 'Failed to download multilingual model'
 
 RUN /app/venv/bin/python -c "from transformers import pipeline; \
-    pipeline('sentiment-analysis', model='siebert/sentiment-roberta-large-english', use_auth_token='$HF_TOKEN')" || echo 'Failed to download model 2'
+    pipeline('sentiment-analysis', model='siebert/sentiment-roberta-large-english', use_auth_token='$HF_TOKEN')" || echo 'Failed to download English model'
 
-# Expose port for FastAPI
+# Expose FastAPI port
 EXPOSE 7860
 
-# Run FastAPI server using virtual environment
-CMD ["/app/venv/bin/uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+# Run FastAPI server using the virtual environment
+CMD ["/app/venv/bin/uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]