Updated Dockerfile for Hugging Face Spaces deployment

Dockerfile

@@ -46,11 +46,13 @@ ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
 
 # Create cache directory
 RUN mkdir -p /app/cache/huggingface
-RUN chmod -R 777 /app/cache/huggingface
+RUN mkdir -p /app/cache/triton
+RUN chmod -R 777 /app/cache
 
 # Set environment variables for cache directory
 ENV HF_HOME=/app/cache/huggingface
 ENV TRANSFORMERS_CACHE=/app/cache/huggingface
+ENV TRITON_CACHE_DIR=/app/cache/triton
 
 # Disable numba caching
 ENV NUMBA_CACHE_DIR=/tmp/numba_cache