Update Dockerfile

Dockerfile

@@ -1,12 +1,15 @@
-# Use a lightweight Python 3.9 image
+# Use a lightweight Python 3.9-slim image
 FROM python:3.9-slim
 
 # 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 a non-root user and switch to it
+# Accept HF_TOKEN as a build argument and set it as an environment variable
+ARG HF_TOKEN
+ENV HF_TOKEN=${HF_TOKEN}
+
+# Create a non-root user and switch to that user
 RUN useradd -m -u 1000 user
 USER user
 
@@ -21,8 +24,8 @@ RUN pip install --no-cache-dir --upgrade pip && \
 # Copy the rest of the application code
 COPY --chown=user . /app
 
-# Expose port 7860 (required by Hugging Face Spaces)
+# Expose port 7860 (Hugging Face Spaces expects this port)
 EXPOSE 7860
 
-# Start the FastAPI app using uvicorn via python -m to ensure it's found
+# Start the FastAPI app using uvicorn via python -m
 CMD ["python", "-m", "uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]