Update Dockerfile

Dockerfile

@@ -1,19 +1,27 @@
+# Use an official Python runtime as a parent image
 FROM python:3.9-slim
 
-# Create a non-root user and switch to it
-RUN useradd -m appuser
+# Set environment variables for Hugging Face cache
+ENV TRANSFORMERS_CACHE=/app/.cache
+ENV HF_DATASETS_CACHE=/app/.cache
+
+# Create the cache directory
+RUN mkdir -p /app/.cache
+
+# Set the working directory in the container
 WORKDIR /app
-RUN chown appuser:appuser /app
-USER appuser
 
 # Copy the requirements file into the container
-COPY --chown=appuser:appuser requirements.txt .
+COPY requirements.txt .
 
 # Install any needed packages specified in requirements.txt
 RUN pip install --no-cache-dir -r requirements.txt
 
 # Copy the current directory contents into the container
-COPY --chown=appuser:appuser . .
+COPY . .
+
+# Pre-download the model during the build process
+RUN python download_model.py
 
 # Expose port 7860 (default port for Gradio)
 EXPOSE 7860