Updated Dockerfile for Hugging Face Spaces deployment

Dockerfile

@@ -44,11 +44,12 @@ ENV PATH=${CUDA_HOME}/bin:${PATH}
 # Optionally set LD_LIBRARY_PATH for CUDA libraries
 ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
 
-# Set environment variable for NeMo cache directory
-ENV NEMO_NLP_TMP=/app/.cache
+# Set environment variable for Hugging Face cache directory
+ENV HF_HOME=/app/.cache/huggingface
+ENV TRANSFORMERS_CACHE=/app/.cache/huggingface
 
 # Create cache directory and set permissions
-RUN mkdir -p /app/.cache && chmod -R 777 /app/.cache
+RUN mkdir -p /app/.cache/huggingface && chmod -R 777 /app/.cache/huggingface
 
 # Copy the setup script and requirements file into the container
 COPY setup.sh requirements.txt /app/