Updated Dockerfile for Hugging Face Spaces deployment

Dockerfile

@@ -47,12 +47,14 @@ ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
 # Create cache directory
 RUN mkdir -p /app/cache/huggingface
 RUN mkdir -p /app/cache/triton
+RUN mkdir -p /app/cache/torch_extensions
 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
+ENV TORCH_EXTENSIONS_DIR=/app/cache/torch_extensions
 
 # Disable numba caching
 ENV NUMBA_CACHE_DIR=/tmp/numba_cache