Update Dockerfile

Dockerfile

@@ -4,9 +4,9 @@ FROM python:3.9
 # Install CUDA for GPU support (optional, based on your needs)
 # Uncomment the following lines if you need GPU support
 # RUN apt-get update && apt-get install -y --no-install-recommends \
- #    cuda-cudart-dev-11.0 \
+#     cuda-cudart-dev-11.0 \
 #   && \
- #   rm -rf /var/lib/apt/lists/*
+#   rm -rf /var/lib/apt/lists/*
 
 # Set the working directory in the container
 WORKDIR /app
@@ -23,5 +23,11 @@ EXPOSE 7860
 # Define environment variable for Gradio server port
 ENV GRADIO_SERVER_PORT=7860
 
+# Set the TRANSFORMERS_CACHE environment variable
+ENV TRANSFORMERS_CACHE=/app/.cache/huggingface/hub
+
+# Create the cache directory if it doesn't exist
+RUN mkdir -p $TRANSFORMERS_CACHE
+
 # Run app.py when the container launches
-CMD ["export TRANSFORMERS_CACHE=/.cache/huggingface/hub","python", "app.py"]
+CMD ["python", "app.py"]