Update Dockerfile

Dockerfile

@@ -1,21 +1,19 @@
-# Use the official Python 3.10 image
 FROM python:3.10
 
-# Set the Hugging Face token as an environment variable
-ARG HUGGINGFACE_HUB_TOKEN
-ENV HUGGINGFACE_HUB_TOKEN=${HUGGINGFACE_HUB_TOKEN}
-
-# Set the working directory inside the container
+# Set working directory
 WORKDIR /app
 
-# Copy the local files into the container's working directory
+# Copy project files
 COPY . /app
 
-# Install the dependencies from requirements.txt
+# Install dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Expose the port FastAPI will use
+# Expose the port that the app will run on
 EXPOSE 7860
 
-# Run the FastAPI application using Uvicorn
+# Set environment variable for Hugging Face cache directory
+ENV HF_HOME="/app/.cache"
+
+# Command to run the FastAPI app with Uvicorn
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]