Update Dockerfile

Dockerfile

@@ -19,10 +19,13 @@ 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 model cache directory exists and is writable
+RUN mkdir -p $HF_HOME && chmod -R 777 $HF_HOME
+
+# Pre-download models
+RUN /app/venv/bin/python -c "from transformers import pipeline; \
+    pipeline('sentiment-analysis', model='johndoee/sentiment')" || echo "Failed to download custom model"
 
-# 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"