Update Dockerfile

Dockerfile

@@ -1,21 +1,24 @@
-# Use an official Python base image
-FROM python:3.11-slim
-
-# Set the working directory inside the container
-WORKDIR /code
-
-# Copy the requirements file and install dependencies
-# We use --index-url for the CPU version of torch to keep it smaller
-RUN echo "torch --index-url https://download.pytorch.org/whl/cpu" > /code/requirements.txt
-COPY ./requirements.txt /code/temp_requirements.txt
-RUN cat /code/temp_requirements.txt >> /code/requirements.txt && rm /code/temp_requirements.txt
-RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
-
-# Copy the rest of the application code
-COPY ./app.py /code/app.py
-
-# Expose the port the app runs on (FastAPI default is 8000)
-EXPOSE 8000
-
-# Command to run the application using uvicorn
+# Use an official Python base image
+FROM python:3.11-slim
+
+# Set the working directory inside the container
+WORKDIR /code
+
+# --- *** ADD THIS LINE *** ---
+# Set an environment variable to tell Hugging Face where to cache models.
+# This avoids permission errors by using a local, writable directory.
+ENV HF_HOME /code/.cache
+# --- ********************* ---
+
+# Copy the requirements file and install dependencies
+COPY ./requirements.txt /code/requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+# Copy the rest of the application code
+COPY ./app.py /code/app.py
+
+# Expose the port the app runs on
+EXPOSE 8000
+
+# Command to run the application
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8000"]