Update Dockerfile

Dockerfile

@@ -1,33 +1,28 @@
-# Use the official Python 3.9 image as a base image
+# Use a lightweight Python 3.9 image
 FROM python:3.9-slim
 
-# Create a new user with UID 1000 and switch to that user
-RUN useradd -m -u 1000 user
-USER user
-
-# Set environment variables for Hugging Face cache and Python
-ENV PATH="/home/user/.local/bin:$PATH"
-ENV TRANSFORMERS_CACHE="/home/user/.cache/huggingface/transformers"
-ENV HF_HOME="/home/user/.cache/huggingface"
+# Set environment variables for Python and Hugging Face cache
 ENV PYTHONUNBUFFERED=1
+ENV TRANSFORMERS_CACHE="/tmp/cache/huggingface/transformers"
+ENV HF_HOME="/tmp/cache/huggingface"
 
-# Create cache directories
-RUN mkdir -p $TRANSFORMERS_CACHE && \
-    mkdir -p $HF_HOME
+# Create a non-root user and switch to it
+RUN useradd -m -u 1000 user
+USER user
 
-# Set the working directory inside the container
+# Set the working directory
 WORKDIR /app
 
-# Copy the requirements file into the container and install dependencies
+# Copy the requirements file and install dependencies
 COPY --chown=user ./requirements.txt /app/requirements.txt
 RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir -r /app/requirements.txt
 
-# Copy all project files into the container
+# Copy the rest of the application code
 COPY --chown=user . /app
 
-# Expose the port that FastAPI will listen on (Hugging Face Spaces requires port 7860)
+# Expose port 7860 (required by Hugging Face Spaces)
 EXPOSE 7860
 
-# Start the FastAPI app using uvicorn, listening on port 7860
+# Start the FastAPI app using uvicorn
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]