Update Dockerfile

Dockerfile

@@ -1,5 +1,3 @@
-# Dockerfile
-
 # Start with a base image that has Python and NVIDIA's CUDA drivers
 FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
 
@@ -15,6 +13,9 @@ RUN apt-get update && apt-get install -y \
 # Set the working directory inside the container
 WORKDIR /app
 
+# Set a writable cache directory for Hugging Face models
+ENV HF_HOME=/app/cache
+
 # Copy and install Python dependencies
 COPY requirements.txt .
 RUN pip3 install --no-cache-dir -r requirements.txt
@@ -26,4 +27,4 @@ COPY main.py .
 EXPOSE 8000
 
 # The command to start the Uvicorn server when the container runs
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]