Update Dockerfile

Dockerfile

@@ -1,21 +1,18 @@
 FROM nvidia/cuda:12.1.0-runtime-ubuntu20.04
 
-# Install system packages
 RUN apt-get update && apt-get install -y \
     python3 python3-pip git && \
     ln -s /usr/bin/python3 /usr/bin/python
 
-# Install Python libraries
 COPY requirements.txt .
 RUN pip install --upgrade pip
 RUN pip install -r requirements.txt
 
-# Copy the app
 WORKDIR /app
+ENV TRANSFORMERS_CACHE=/app/cache
+RUN mkdir -p /app/cache
+
 COPY app.py .
 
-# Expose the port required by Hugging Face Spaces or your own server
 EXPOSE 7860
-
-# Run FastAPI app
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]