Update Dockerfile

Dockerfile

@@ -1,10 +1,27 @@
+# Use an official Python runtime as a parent image
 FROM python:3.9
 
+# Set the working directory in the container
 WORKDIR /app
+
+# Copy the application files to the container
 COPY . /app
 
+# Install dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Set environment variables
+ENV HF_HOME="/tmp/huggingface"
+ENV HF_DATASETS_CACHE="/tmp/huggingface/datasets"
+ENV TRANSFORMERS_CACHE="/tmp/huggingface/transformers"
+ENV HF_HUB_CACHE="/tmp/huggingface/hub"
+
+# Ensure cache directories exist and are writable
+RUN mkdir -p /tmp/huggingface/datasets /tmp/huggingface/transformers /tmp/huggingface/hub
+RUN chmod -R 777 /tmp/huggingface  # Give full permissions
+
+# Expose the application port
 EXPOSE 7860
 
-CMD ["gunicorn", "-b", "0.0.0.0:7860", "app:app"]
+# Start the Flask app with Gunicorn
+CMD ["gunicorn", "-w", "1", "-b", "0.0.0.0:7860", "app:app"]